Cleaned up TODO
Updated TODO