Added remove functionality.