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

        button.setTitle("TEST", for: .normal)
        button.setTitleColor(.menuBackground, for: .normal)
        button.backgroundColor = .menuButton
        button.translatesAutoresizingMaskIntoConstraints = false
        
        return button
    }()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        view.backgroundColor = .menuBackground
        
        setupViews()
    }
    

    fileprivate func setupViews() {
        setupTrainButton()
        setupTestButton()
        setupNeuronViews()
    }
    
    fileprivate func setupNeuronViews() {
        
        for viewIndex in 0..<6 {
            
            let neuronView = UIView()
            var verticalOffset: CGFloat = 3
            var horizontalOffset = -self.view.frame.width/2
            var isOnTop = true
            
            neuronView.backgroundColor = .menuButton
            
            if viewIndex >= 3 {
                horizontalOffset = horizontalOffset * -1
            }
            
            if viewIndex == 1 || viewIndex == 4 {
                verticalOffset = self.view.frame.height
            }
            
            if viewIndex == 2 || viewIndex == 5 {
                isOnTop = false
            }
            
            
            setupConstraintsForMenu(view: neuronView, isOnTop: isOnTop, verticalOffset: verticalOffset, horizontalOffset: horizontalOffset, width: 55)
            closeNeuronViews.append(neuronView)
        }
    }
    
    fileprivate func drawSynapsePathsFor(view: UIView) {
        
        // ??
        let viewCenter = CGPoint(x: view.frame.midX, y: view.frame.midY)
        let buttonCenter = CGPoint(x: trainButton.frame.midX, y: trainButton.frame.midY)
        
        let path = UIBezierPath()
        path.move(to: viewCenter)
        path.addLine(to: buttonCenter)
        path.close()
        
        let layer = CAShapeLayer()
        layer.path = path.cgPath
        layer.strokeColor = UIColor.black.cgColor
        layer.lineWidth = 5.0
        
        view.layer.addSublayer(layer)
    }
    
    fileprivate func setupTrainButton() {
        setupConstraintsForMenu(view: trainButton, isOnTop: true, verticalOffset: 5, horizontalOffset: 0, width: 75)
    }
    
    fileprivate func setupTestButton() {
        setupConstraintsForMenu(view: testButton, isOnTop: false, verticalOffset: 5, horizontalOffset: 0, width: 75)
    }
    
    fileprivate func setCornerRadiusAndShadowFor(view: UIView) {
