list-comprehensions on hackerrank