Add Stop button functionality