add validation in models