diff --git a/TORoundedButtonExample/Base.lproj/Main.storyboard b/TORoundedButtonExample/Base.lproj/Main.storyboard
index 9bb8f0a..c555c26 100644
--- a/TORoundedButtonExample/Base.lproj/Main.storyboard
+++ b/TORoundedButtonExample/Base.lproj/Main.storyboard
@@ -17,13 +17,10 @@
-
+
+
-
-
-
-
@@ -41,9 +38,7 @@
-
-
diff --git a/TORoundedButtonExample/ViewController.m b/TORoundedButtonExample/ViewController.m
index 37fec05..2b2843c 100644
--- a/TORoundedButtonExample/ViewController.m
+++ b/TORoundedButtonExample/ViewController.m
@@ -30,6 +30,14 @@ - (void)viewDidLoad {
self.button.tappedHandler = ^{
[weakSelf playFadeAnimationOnView:weakSelf.tappedLabel];
};
+
+ // Uncomment to have the button shrink to wrap the text
+ // [self.button sizeToFit];
+}
+
+- (void)viewDidLayoutSubviews {
+ [super viewDidLayoutSubviews];
+ self.button.center = self.view.center;
}
- (void)playFadeAnimationOnView:(UIView *)view