Add remove functionality.