Update Unit.java with getters/setters