updated unitClasses diagram