[iterators][flatten_iterator] change statement layout to be nicer