Update a TODO
Fix documentation