Misc fixes + update TODO