Turning a for loop into a list comprehension

https://www.pythonmorsels.com/turning-loop-list-comprehension/