Move setup to import_census.R and focus on model