Updated UI remove function

Now update children hierarchy