Add return and exception expectations