Finish assignment 3