updated uml diagram visuals