Add necessary TODOs