Remove imports in functions