Add model validation