         userContentController.addUserScript(script)
         let configuration = WKWebViewConfiguration()
         configuration.userContentController = userContentController;
-        return WKWebView(frame: self.view.bounds, configuration: configuration)
+        return WKWebView(frame: view.bounds, configuration: configuration)
     }
     
     private func getSource() -> String {
-        let bundle = NSBundle.mainBundle()
+        let bundle                  = NSBundle.mainBundle()
         let playlistifyPath: String = bundle.pathForResource("playlistify-userscript", ofType: "js")!
         let mainPath:        String = bundle.pathForResource("main", ofType: "js")!
-        let source = String(contentsOfFile: playlistifyPath)! + String(contentsOfFile: mainPath)!
-        return source
+        return String(contentsOfFile: playlistifyPath)! + String(contentsOfFile: mainPath)!
     }
     
     func updateViews() {


    func loadURL(url: NSURL) {
        currentURL = url
        if let webView = self.webView {
            webView.loadRequest(NSURLRequest(URL: url))
        }
    }



    override func viewDidLoad() {
        super.viewDidLoad()
        webView = createWebView()
        webView!.setTranslatesAutoresizingMaskIntoConstraints(false)
        view.insertSubview(webView!, atIndex:0)
        view.addConstraints([
            NSLayoutConstraint(item: webView!,
                attribute: NSLayoutAttribute.Width,
                relatedBy: NSLayoutRelation.Equal,
                   toItem: view,
                attribute: NSLayoutAttribute.Width,
               multiplier: 1.0,
                 constant: 0),
            NSLayoutConstraint(item: webView!,
                attribute: NSLayoutAttribute.Height,
                relatedBy: NSLayoutRelation.Equal,
                   toItem: view,
                attribute: NSLayoutAttribute.Height,
               multiplier: 1.0,
                 constant: 0)
            ])
        
        webView!.allowsBackForwardNavigationGestures = true
        playlistButton        = UIBarButtonItem(image: UIImage(named: "playlist"),
                                                style: UIBarButtonItemStyle.Plain,
                                               target: self,
