Add ability to update the map