more assignment 4 corrections