Updates TODO and other docs