Added support for specify_class

changes only relating the specify_class