class diagram, small changes