Add TODO comment about other steps that could be added