add decorator class