Modify the assignement01.java