Update TODO docs