What's wrong with my code? (If's and remembering choices