Update puzzlesolver.py