Add validation to User model