             attributedSelectedTitle = friendSelectedAttributedTitle
             normalBackgroundColor = establishedNormalBackgroundColor
             selectedBackgroundColor = establishedSelectedBackgroundColor
+            borderColor = .blackColor()
         case .Noise:
             attributedNormalTitle = noiseNormalAttributedTitle
             attributedSelectedTitle = noiseSelectedAttributedTitle
             normalBackgroundColor = establishedNormalBackgroundColor
             selectedBackgroundColor = establishedSelectedBackgroundColor
+            borderColor = .blackColor()
         case .Mute:
             attributedNormalTitle = muteNormalAttributedTitle
             attributedSelectedTitle = muteSelectedAttributedTitle
             normalBackgroundColor = muteNormalBackgroundColor
             selectedBackgroundColor = muteSelectedBackgroundColor
+            borderColor = .redColor()
         default:
             attributedNormalTitle = followNormalAttributedTitle
             attributedSelectedTitle = followSelectedAttributedTitle
             normalBackgroundColor = followNormalBackgroundColor
             selectedBackgroundColor = followSelectedBackgroundColor
+            borderColor = .blackColor()
         }
         label.attributedText = active ? attributedSelectedTitle : attributedNormalTitle
         mainButtonBackground.backgroundColor = active ? selectedBackgroundColor : normalBackgroundColor
-
+        mainButtonBackground.layer.borderColor = borderColor.CGColor
         updateLayout()
     }
 

        moreButton.hidden = true
        attributedNormalTitle = followNormalAttributedTitle
        attributedSelectedTitle = followSelectedAttributedTitle
        mainButtonBackground.layer.borderColor = UIColor.blackColor().CGColor
        mainButtonBackground.layer.borderWidth = 1
    }


        attributed.addAttributes(attributes, range: range)
        return attributed
    }
}
