Update model to include model attributes