Support return comment on newline.

This satisfy 'Scenario: handles return comment on newline'.

`<code> #=> result` is nice for short ones, but become less clear when
test code or result is long.