diff --git a/PSTreeGraphView/PSBaseSubtreeView.m b/PSTreeGraphView/PSBaseSubtreeView.m
index 74b4728..5cdd5af 100644
--- a/PSTreeGraphView/PSBaseSubtreeView.m
+++ b/PSTreeGraphView/PSBaseSubtreeView.m
@@ -23,11 +23,13 @@
 
 #define CENTER_COLLAPSED_SUBTREE_ROOT   1
 
-static UIColor *subtreeBorderColor(void) {
+static UIColor *subtreeBorderColor(void) 
+{
     return [[UIColor colorWithRed:0.0 green:0.5 blue:0.0 alpha:1.0] retain];
 }
 
-static CGFloat subtreeBorderWidth(void) {
+static CGFloat subtreeBorderWidth(void) 
+{
     return 2.0;
 }
 
@@ -41,7 +43,7 @@ @implementation PSBaseSubtreeView
 @synthesize nodeView;
 
 
-- (BOOL)isLeaf 
+- (BOOL) isLeaf 
 {
     return [[[self modelNode] childModelNodes] count] == 0;
 }
diff --git a/PSTreeGraphView/PSBaseTreeGraphView.h b/PSTreeGraphView/PSBaseTreeGraphView.h
index 33e9c13..bc547ea 100644
--- a/PSTreeGraphView/PSBaseTreeGraphView.h
+++ b/PSTreeGraphView/PSBaseTreeGraphView.h
@@ -203,7 +203,7 @@ typedef NSInteger PSTreeGraphOrientationStyle;
 
 // The style for tree graph orientation  (See the TreeGraphOrientationStyle enumeration above.)
 
-@property (nonatomic) PSTreeGraphOrientationStyle treeGraphOrientation;
+@property (nonatomic, assign) PSTreeGraphOrientationStyle treeGraphOrientation;
 
 //Returns YES if the tree needs relayout.
  
@@ -295,7 +295,7 @@ typedef NSInteger PSTreeGraphOrientationStyle;
 
 // The style for node connecting lines.  (See the PSTreeGraphConnectingLineStyle enumeration above.)
  
-@property (nonatomic) PSTreeGraphConnectingLineStyle connectingLineStyle;
+@property (nonatomic, assign) PSTreeGraphConnectingLineStyle connectingLineStyle;
 
 // Defaults to NO.  If YES, a stroked outline is shown around each of the TreeGraph's 
 // SubtreeViews.  This can be helpful for visualizing the TreeGraph's structure and layout.