Finish assignment 5