python list comprehension hackerrank