Updated unit uml in diagram