Use dictionary comprehension