Define List Comprehension In Python Hackerrank