Lecture 2 assignment 2 finished