Update bouncing-ball.py