Working on Assignment 6