Add TODO comment to Model