updating changes for assignment 4