Adding Assignment 4