added test in list comprehension