Add TODO note to code