Misc changes to jax lectures#533

jstac merged 3 commits intomainQuantEcon/lecture-python-programming:mainfrom

jxmQuantEcon/lecture-python-programming:jxmCopy head branch name to clipboard