Create balls.py