Add TODO for ifs in comprehensions