Updated puzzle.c