Update Unit Controller, model and views

POS