added class-diagram