https://blog.soshace.com/comprehension-in-python/