updated unit.h.

Will be doing more to properly make use of open/closed principal over the next couple days.