Add component removal from entities