Changes to make the balls move again!