diff --git a/contracts/docs/plantuml/sonicContracts.png b/contracts/docs/plantuml/sonicContracts.png
index 0241cdefc5..78ea2c39d4 100644
Binary files a/contracts/docs/plantuml/sonicContracts.png and b/contracts/docs/plantuml/sonicContracts.png differ
diff --git a/contracts/docs/plantuml/sonicContracts.puml b/contracts/docs/plantuml/sonicContracts.puml
index 6946244815..29c25d2d2e 100644
--- a/contracts/docs/plantuml/sonicContracts.puml
+++ b/contracts/docs/plantuml/sonicContracts.puml
@@ -1,17 +1,19 @@
 @startuml
 
 !$originColor = DeepSkyBlue
+!$phase2 = Yellow
 ' !$originColor = WhiteSmoke
 !$newColor = LightGreen
 !$changedColor = Orange
 !$thirdPartyColor = WhiteSmoke
 
-' legend
-' blue - Origin
-' ' green - new
-' ' orange - changed
-' white - 3rd Party
-' end legend
+legend
+blue - Origin
+' green - new
+' orange - changed
+yellow - phase2
+white - 3rd Party
+end legend
 
 title "Sonic Contract Dependencies"
 
@@ -49,10 +51,14 @@ object "Special Fee Contract" as sfc <<Sonic>><<Proxy>> {
     asset: S
 }
 
-' object "AMOStrategy" as amoStrat <<Origin>><<Proxy>> #$originColor {
-'     asset: wS
-'     reward: ?
-' }
+object "AMOStrategy" as amoStrat <<Origin>><<Proxy>> #$phase2 {
+    asset: wS
+    reward: ?
+}
+
+object "OSonicHarvester" as harv <<Origin>><<Proxy>> #$phase2 {
+    rewards: ?
+}
 
 wos <. zap
 zap ..> os
@@ -63,9 +69,14 @@ checker ..> vault
 wos ..> os
 os <.> vault
 vault <.> drip
-vault <..> stakeStrat
+vault <...> stakeStrat
 stakeStrat ..> sfc
-' vault <...> amoStrat
+vault <...> amoStrat
+
+vault <.. harv 
+drip <.. harv 
+
+harv <..> amoStrat
 
 
 @enduml
\ No newline at end of file