List Comprehension Python Hackerrank