Took out some prints so you can just watch the reward, added headless training if desired, fixed a mistake in my previous changes to the reward function