Tidy up TODOs