Added support for use special class for controllers