Added Assignment 4 work from video lecture