https://myeduwaves.com/python-solution/write-a-function-hackerrank-python-solution/