Does this correctly handle stuff like:

```
object Test extends App {
  def foo: Int = {
    List(1, 2).foreach(_ => return 1)
    2
  }
  println(foo)
}
```