add onPieceRemoved method to unitController