some fixes to uml class diagram