More TODO updates