Minor changes. Updated TODO.