Assignment 2 DONE