https://slothcoders.blogspot.com/2021/02/write-a--function-in-python-solution.html