Updated TODO items
More TODO tweaks