Updated the model to reflect new columns