Add Joi validation