Update puzzle.ml