add auto stop logic