Update Project 4: make some changes