updated remove method