Added entity removal button