Update TODO
More documentation