Reverting old file
Changed the wrong line
Readded dict comprehension