Update Falling Balls.py