Updated it to use the map div properly