added class design