Change base class for controllers