Engine: strip automatically `return` when in end position

Any `return` statement currently triggers the monadic rewrite phase, which lifts everything to a monad.

When a `return` statement is in end position, we should instead just drop it.