Updates for assignment 4