Added code to display the ball