fix up lecture 16