Added functionality for adding units in phase 1