Improve TODOs
Improve documentation