Update puzzle.cpp