added decorator as a class