Done with assignment 2