Add TODO notes