Add attribute, change base class for controller