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