Update Puzzle.py