Add stop functionality