Currently, it seems like the observation space only contains v_y. Having v_x or tangential velocity might be better.