Update UML diagrams to reflect recent code changes