Updating data import script to work with new changes in application