Changes made for assignment 2