10. Adding validation

10. Adding validation