Update AI and TODO docs