minor updates

Author: HumphreyYang

lectures/_toc.yml

@@ -79,7 +79,6 @@ parts:
   - file: cass_fiscal
   - file: cass_fiscal_2
   - file: ak2
-  - file: ak_aiyagari
   - file: cake_eating_problem
   - file: cake_eating_numerical
   - file: optgrowth
@@ -106,6 +105,7 @@ parts:
   - file: markov_perf
   - file: uncertainty_traps
   - file: aiyagari
+  - file: ak_aiyagari
 - caption: Asset Pricing and Finance
   numbered: true
   chapters:

lectures/ak_aiyagari.md

@@ -734,7 +734,7 @@ ss1 = find_ss(hh, firm, [0, 0.1, np.zeros(hh.j_grid.size)], Q, verbose=True)
 ```
 
 ```{code-cell} ipython3
-%time find_ss(hh, firm, [0, 0.1, np.zeros(hh.j_grid.size)], Q);
+%time find_ss(hh, firm, [0, 0.1, np.zeros(hh.j_grid.size)], Q).block_until_ready();
 ```
 
 ```{code-cell} ipython3