Add decorator helpers