added class code