Minor fixes
Updated TODO