https://remp3indir.net/python-write-a-function-problem-introduction-in-hackerrank/