Task B1. Add a validation