fixed UML diagram, hopefully