Assignment 2 of lecture 3