Updated TODO a bit