Finished with assignment 2