Add TODO for badges