Added new puzzle code