https://videosashka.com/watch/kfaPmWceI_M/python-tutorial-5-write-a-function-hackerrank-solution