Assignment 4 Modified