updating the code and dataset for changes in column names