Updated data import script