* Fix compile errors

* Update TODO