diff --git a/OGSwitch.xcworkspace/xcuserdata/oskar.xcuserdatad/UserInterfaceState.xcuserstate b/OGSwitch.xcworkspace/xcuserdata/oskar.xcuserdatad/UserInterfaceState.xcuserstate
index e48d4ba..f2407e8 100644
Binary files a/OGSwitch.xcworkspace/xcuserdata/oskar.xcuserdatad/UserInterfaceState.xcuserstate and b/OGSwitch.xcworkspace/xcuserdata/oskar.xcuserdatad/UserInterfaceState.xcuserstate differ
diff --git a/OGSwitch/OGSwitch.xcodeproj/project.pbxproj b/OGSwitch/OGSwitch.xcodeproj/project.pbxproj
index 4bf455d..66ed6ca 100644
--- a/OGSwitch/OGSwitch.xcodeproj/project.pbxproj
+++ b/OGSwitch/OGSwitch.xcodeproj/project.pbxproj
@@ -7,9 +7,9 @@
 	objects = {
 
 /* Begin PBXBuildFile section */
-		75256FD820FA5A0E00ED6782 /* OGSwitch.h in Headers */ = {isa = PBXBuildFile; fileRef = 75256FD620FA5A0E00ED6782 /* OGSwitch.h */; settings = {ATTRIBUTES = (Public, ); }; };
 		75256FDF20FA5A3E00ED6782 /* OGSwitch.swift in Sources */ = {isa = PBXBuildFile; fileRef = 75256FDE20FA5A3E00ED6782 /* OGSwitch.swift */; };
-		753FF19F20FA621C00DFA33E /* OGSwitch.swift in Headers */ = {isa = PBXBuildFile; fileRef = 75256FDE20FA5A3E00ED6782 /* OGSwitch.swift */; settings = {ATTRIBUTES = (Public, ); }; };
+		753FF1A520FA64B500DFA33E /* OGSwitch.swift in Headers */ = {isa = PBXBuildFile; fileRef = 75256FDE20FA5A3E00ED6782 /* OGSwitch.swift */; settings = {ATTRIBUTES = (Public, ); }; };
+		753FF1A620FA64D200DFA33E /* OGSwitch.h in Headers */ = {isa = PBXBuildFile; fileRef = 75256FD620FA5A0E00ED6782 /* OGSwitch.h */; settings = {ATTRIBUTES = (Public, ); }; };
 /* End PBXBuildFile section */
 
 /* Begin PBXFileReference section */
@@ -63,8 +63,8 @@
 			isa = PBXHeadersBuildPhase;
 			buildActionMask = 2147483647;
 			files = (
-				75256FD820FA5A0E00ED6782 /* OGSwitch.h in Headers */,
-				753FF19F20FA621C00DFA33E /* OGSwitch.swift in Headers */,
+				753FF1A620FA64D200DFA33E /* OGSwitch.h in Headers */,
+				753FF1A520FA64B500DFA33E /* OGSwitch.swift in Headers */,
 			);
 			runOnlyForDeploymentPostprocessing = 0;
 		};
diff --git a/OGSwitch/OGSwitch/OGSwitch.h b/OGSwitch/OGSwitch/OGSwitch.h
index 2cace5b..fe3bb83 100644
--- a/OGSwitch/OGSwitch/OGSwitch.h
+++ b/OGSwitch/OGSwitch/OGSwitch.h
@@ -13,7 +13,3 @@ FOUNDATION_EXPORT double OGSwitchVersionNumber;
 
 //! Project version string for OGSwitch.
 FOUNDATION_EXPORT const unsigned char OGSwitchVersionString[];
-
-// In this header, you should import all the public headers of your framework using statements like #import <OGSwitch/PublicHeader.h>
-
-
diff --git a/OGSwitch/OGSwitch/OGSwitch.swift b/OGSwitch/OGSwitch/OGSwitch.swift
index 69a19e4..83164c8 100644
--- a/OGSwitch/OGSwitch/OGSwitch.swift
+++ b/OGSwitch/OGSwitch/OGSwitch.swift
@@ -11,7 +11,7 @@ import QuartzCore
 
 //Can not be NSControl due to 10.9 issues: https://github.com/iluuu1994/ITSwitch/issues/21
 
-public class OGSwitch : NSView {
+@IBDesignable public class OGSwitch : NSView {
     
     @IBInspectable public var tintColor: NSColor = NSColor(calibratedRed:0.27, green: 0.86, blue: 0.36, alpha: 1.0) {
         didSet {