Adds block returns for `and_return`