fix: add training dir to sys.path so -m training.test_rollout works on Colab

fix: colab working dir bug, rollout sys.path, openenv imports, add plot_rewards

first commit