changed Problem 1 using dictionary comprehension