removeUnit() function added