Create python_puzzle.py