Help Me Solve Python Hackerrank Interview