Fixing Assignment Four