Updated Map function