better flow for comprehensions