updating Assignment 02