add remove functionality

Closes #1