Finish assignment 1