Finish assignment
Done with assignment 2