Added more to the model