- removeUnit properly working now