add testing remove function