https://remp3indir.net/python-hackerrank-solutions-write-a-function-easy-problem/