I added validation to the model