         return flashButton
     }()
     open var cameraSwitchButton: UIButton!
-    
+    open var captureButton: UIButton!
     
     override open func viewDidLoad() {
         super.viewDidLoad()
         
         self.setupDevices()
         self.setupUI()
-        self.beginSession()
+        self.setupSession()
         
         self.setupMotionManager()
     }

        contentView.addGestureRecognizer(UITapGestureRecognizer(target: self, action: #selector(DKCamera.handleFocus(_:))))
    }
    
    // MARK: - Callbacks
    
    internal func dismiss() {

        return true
    }
    
    // MARK: - Setup
    
    let bottomView = UIView()
    open func setupUI() {
        self.view.backgroundColor = UIColor.black
        self.view.addSubview(self.contentView)
