Assignment 6 continued