modified uml diagram