use a dictionary comprehension
add a ghetto unit test