Maintain newlines in comprehension translation a little better.

Newline tracking is still not perfect, but probably doesn't need to be for comprehensions, which generally are written on one line.