Updated UnitDiagrams

Matches .pu code now