How would I write this entire code into a single return statement? I was able to get the second half into a list comprehension, but not sure how to get the first half into a
