add class Attribute, but need to fix compile error