Add decorator for controllers