Add todo to code