Update assignment-2.java