update Assignment2