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

first commit