Added remove function, need to fix bug with nodeToDelete