Update puzzles.py