Update Unit.java

Accidentally replaced with the wrong .java file.