Made minor changes to diagram