Fixed up TODOs