I tried it out for advancedChoiceWorld and everything works great. I also defined ‘STIM_GAIN’ in the trainingChoiceWorld task_parameters.yaml file which did reverse the wheel contingency. However, the program still rewards as if it was the original task (e.g. reward given when the visual stimulus was moved out of the screen, and time out when moved to the centre).