Add doc TODO comment