Kind of finish Assignment2.java