Changes to import script