Added Assignment 5 code