Make some minor changes on Project 4