 
   preparePreview: (preview, callback) ->
     @runtime.connect preview
+    @$connButton.click =>
+      @preparePreview preview, ->
+
     @runtime.once 'connected', =>
       @$connector.find('h2 span').html @runtime.getAddress()
+      @$connButton.hide()
+      @$startButton.show()
       callback()
+    @runtime.once 'disconnected', =>
+      @$connButton.show()
+      @$startButton.hide()
+      @$stopButton.hide()
 
 plugin = Dataflow::plugin 'preview'
 Dataflow::plugins['preview'] = new NoFloPreview

      <div class=\"toolbar\">
        <button class=\"start btn\"><i class=\"icon-play\"></i></button>
        <button class=\"stop btn\"><i class=\"icon-stop\"></i></button>
        <span class=\"status\"></span>
        <span class=\"uptime\"></span>
      </div>

    @$preview = @$connector.find '.preview'
    @$startButton = @$connector.find '.start'
    @$stopButton = @$connector.find '.stop'
    @$connButton = @$connector.find '.connect'

    dataflow.addPlugin
      id: 'preview'
