Updates to cartpole system to get it to run