Improve the sequence diagram