https://techbeamers.com/python-list-comprehension/