Assignment 4 update