Bouncing Ball Program In Python