Refactor TODOs