Add registerClass support