updating Assignment2 UI