Just finished this, need a part 2 for the PPO reward functionality