You can still get rid of `return`