ice cream parlor hackerrank solution in python