Update the .puml code