cleaned up 20, swapped a for loop for list comprehension

you should be proud emil!