Added a decorator for classes.