Update to Class Diagram and the way program operates