add small changes with list comprehensions