Adjust to new solution setup
@@ -410,9 +410,9 @@ periods = range(t_end - t_start + 1)
n_slices = [0,2,6]
mMax = 20
-DFuncAdj = [ContribAgent.solution[t].RebStage.DFuncAdj for t in periods]
-ShareFuncSha = [ContribAgent.solution[t].ShaStage.ShareFuncAdj for t in periods]
-cFuncFxd = [ContribAgent.solution[t].ConStage.cFunc for t in periods]
+DFuncAdj = [ContribAgent.solution[t]['Reb'].DFuncAdj for t in periods]
+ShareFuncSha = [ContribAgent.solution[t]['Sha'].ShareFuncAdj for t in periods]
+cFuncFxd = [ContribAgent.solution[t]['Con'].cFunc for t in periods]
# Create projected consumption functions at different points of the share grid
shares = [0., 0.2]
