add decorator using class