Implemented the removal of units post base attack.