Replace all instances of `return` with `pure`

Due to the accepted "monad of no return" proposal, return becomes an
alias for pure. Return can be a pitfall for newcomers, so we decided to
use pure instead.