Prepared some code for the "Puzzle" task.