Add working flat_map