updating Assignment 2