Added in the ball movement logic, about to test it