Finish assignment 4 and 5