Add @base class