Assignment 4 continued