add puzzle.py file