Finished assignment Reinforcement Learning