Update Project 4: Fix some problems