updated TODO comment