Adding option to run DataSetCounter as sink