My hackerrank.com code