Add handling of RemoveEntityCommand