Now that it's applied to the whole function, I wonder if there is a decorator for that.