Add TODO / documentation