Update HW2
lecture 5