Assignment 4 corrections