Minor. Update TODOs