Updated TODO to match latest commit