Made UML diagram match actual code.