More changes to diagram.