adding remove functionality