Updates to the reimport script