Assignment2 update