Update unit.h