Hackerearth Practice Python