Python Hacker Rank Problem