Solving hackerrank problems with python