Update balls.py