minor fixes to match lesson