     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) {

        view.layer.shadowOpacity = 0.5
    }
    
    fileprivate func setupConstraintsForMenu(view: UIView, isOnTop: Bool, verticalOffset: CGFloat, horizontalOffset: CGFloat, width: CGFloat) {
        
        self.view.addSubview(view)
        
        var verticalOffsetConstant = self.view.frame.height / verticalOffset
        
        verticalOffsetConstant = isOnTop ? -verticalOffsetConstant : verticalOffsetConstant
        
        let horizontalConstraint = NSLayoutConstraint(item: view, attribute: .centerX, relatedBy: .equal, toItem: self.view, attribute: .centerX, multiplier: 1, constant: horizontalOffset)
        let verticalConstraint = NSLayoutConstraint(item: view, attribute: .centerY, relatedBy: .equal, toItem: self.view, attribute: .centerY, multiplier: 1, constant: verticalOffsetConstant)
        
        self.view.addConstraint(horizontalConstraint)
        self.view.addConstraint(verticalConstraint)
        self.view.addConstraintsWithFormat(format: "H:[v0(\(width))]", views: view)
        self.view.addConstraintsWithFormat(format: "V:[v0(\(width))]", views: view)
        
        horizontalConstraint.isActive = true
        verticalConstraint.isActive = true
