Add support for type parameters for classes