Remove comments
Use list comprehension