Add ability to assign weights for each unit.