Rewrite the first map as a for comprehension