Added run/stop button functionality