UPDATE: main_training_reward_based_new.py runs now with with one test -- evaluation not done yet