Update TODO
Minor style changes