updated remove function