Updated map function