https://remp3indir.net/6-write-a-function-hackerrank-python-solution/