Add decorator functionality