small final tweaks to lecture 21