Adjust browse diagram code