Add support for Trait attributes on classes in addition to methods.