Assignment 2 first part