Adding arguments to RL_trials