My hackerrank code