Puzzle Problem In Python