https://slothcoders.blogspot.com/2021/02/list-comprehensions-in-python-solution.html