Added callback to update the map