Remove some TODOs