refactor: use list comprehension