Added remove function