Assignment 4 adjustments