Added Assignment 6 code