Fixed some minor mistakes in the model