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

    fileprivate func setupViews() {
        setupTrainButton()
        setupTestButton()
    }
    
    fileprivate func setupTrainButton() {
        setupConstraintsForMenu(button: trainButton, isOnTop: true)
    }
    
    fileprivate func setupTestButton() {
        setupConstraintsForMenu(button: testButton, isOnTop: false)
    }
    
    fileprivate func setCornerRadiusAndShadowFor(view: UIView) {

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