Add assertion
Add UML TODOs