Python Hackerearth Practice