Finish assignment 4