Add callback arg for stop()