Write A Function In Python Hackerrank