Add validations to model