Add support for class attributes