[olg] modify code

[shlff]

          modify code

• make this a function

  lecture-python-intro/lectures/olg.md

  Line 311 in 94272e4

  ax.plot(R_vals, np.ones_like(R_vals) * (β / (1 + β)) * w,

Originally posted by @shlff in #312 (comment)