Add the map function