Added the initialization part of the diagram