Add decorator generator