Updated Map code