Thanks! I have updated my post to use your code. It is indeed much nicer. And yes, I don't write much FP. I just improved the comprehension code as well using

Thanks! I have updated my post to use your code. It is indeed much nicer. And yes, I don't write much FP. I just improved the comprehension code as well using the same idea as your code, eliminating an entire list!