minor change to diagram