added a setter for unit