Adding remove() function