Adds the "remove" option to the set method

Fixes [MLHR-1410](https://malhar.atlassian.net/browse/MLHR-1410)