Trying to get the map to work correctly