Sequence diagram generation enhancement