improve on import script