minor TODO changes