Adding support for stop_n_workers.
Updated docs