Removed duplicate lines and added 'Class Diagram'