Added UnitSetup class for adding units in gui