Create simple-balls.py