A simple memorization game [Python]