Add some validation to my model