Add validations to some models