Add return handler