Puzzle some code added