Unit.h changes