More TODO changes