Add TODO to cleanup copied code.