Add map.map.flatMap test