         button.setTitle("TEST", for: .normal)
         button.setTitleColor(.menuBackground, for: .normal)
         button.backgroundColor = .menuButton
-        button.translatesAutoresizingMaskIntoConstraints = false
         
         return button
     }()
     
+    var closeNeuronViews: [UIView] = []
+    var distantNeuronViews: [UIView] = []
+    
+    // MARK: - Lifecycle -
+    
     override func viewDidLoad() {
         super.viewDidLoad()
+        
         view.backgroundColor = .menuBackground
-        
         setupViews()
     }
     

    let trainButton: UIButton = {
        let button = UIButton()
        button.setTitle("TRAIN", for: .normal)
//        button.setTitleColor(.menuButtonPurple, for: .normal)
        button.backgroundColor = .menuButton
        button.translatesAutoresizingMaskIntoConstraints = false
    

        
        setCornerRadiusAndShadowFor(view: trainButton)
        setCornerRadiusAndShadowFor(view: testButton)
        
        for view in closeNeuronViews {
            setCornerRadiusAndShadowFor(view: view)
            drawSynapsePathsFor(view: view)
        }
    }
    
    // MARK: - Helpers -
