Create 2balls.py