Add TODO comment section