If that doesn't quite work, we could also probably use after do to attach it to some method calls