improve import script