Add @class comment