Assignment3 submission

I was able to change it to a dictionary comprehension.