Update Unit.java