minor changes for new diagram