Assignment 1 - update 4