Modified Class Diagram. Looked/Changed some code.