Updated diagram
Minor comment adjustments