Changed to nest list comprehension cuz we're cool