Adding remove function