Edit the unit