         self.navigationController?.navigationBar.setBackgroundImage(UIImage(), for: UIBarMetrics.default)
         self.navigationController?.navigationBar.shadowImage = UIImage()
         
+        //set so screen does not lock while drumming
+        UIApplication.shared.isIdleTimerDisabled = true
     }
     
     override func viewDidAppear(_ animated: Bool) {

        
        drumKits = loadDrumKits()
        selectedDrumKit = loadSelectedKit()
        
        
        
        doMachineLearning()

    func finishRecording() {
        audioRecorder.stop()
        audioRecorder = nil
    }
    
    
    func doMachineLearning() {
