return block with argument which gets passed along in Block#after: & Block#then:
add additional test case for Block#arity