Create puzzle.py