https://remp3indir.net/function-in-python-hackerrank-practice-question/