Update unit.cpp