Updated TODO note