Updated TODO

Updated TODO