Added remove functionality