Lecture 2 Review

fix bug