Update Unit.java to have mutator methods