tried to fix the error with the columns in pandas, not sure it is completely correct