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