Assignment 2 final draft without comments