The warmup hackerrank problems in python