https://mixrolikus.cc/video/FP9ZuA6tX3g/using-if-else-in-python-list-comprehension