https://freecodecenter.com/for-loop-example-in-python/