add one line alternative with list comprehension. It does not seem readable for me nonetheless