Add a TODO
Improve validation