Updated TODO

: restructure