3 files changed, 15 insertions, 7 deletions
diff --git a/OLMKit.podspec b/OLMKit.podspec
index 133e23d..f843f93 100644
--- a/OLMKit.podspec
+++ b/OLMKit.podspec
@@ -32,6 +32,7 @@ Pod::Spec.new do |s|
   }
 
   s.source_files = "xcode/OLMKit/*.{h,m}", "include/**/*.{h,hh}", "src/*.{c,cpp}", "lib/crypto-algorithms/sha256.c",  "lib/crypto-algorithms/aes.c", "lib/curve25519-donna/curve25519-donna.c"
+  s.private_header_files = "xcode/OLMKit/*_Private.h"
 
   # Those files (including .c) are included by ed25519.c. We do not want to compile them twice
   s.preserve_paths = "lib/ed25519/**/*.{h,c}"
diff --git a/xcode/OLMKit/OLMKit.h b/xcode/OLMKit/OLMKit.h
index e23a9f1..455d11b 100644
--- a/xcode/OLMKit/OLMKit.h
+++ b/xcode/OLMKit/OLMKit.h
@@ -18,10 +18,6 @@
 
 #import <Foundation/Foundation.h>
 
-
-//! Project version string for OLMKit, the same as libolm.
-NSString *OLMKitVersionString();
-
 // In this header, you should import all the public headers of your framework using statements like #import <OLMKit/PublicHeader.h>
 
 #import <OLMKit/OLMAccount.h>
@@ -30,3 +26,10 @@ NSString *OLMKitVersionString();
 #import <OLMKit/OLMUtility.h>
 #import <OLMKit/OLMInboundGroupSession.h>
 #import <OLMKit/OLMOutboundGroupSession.h>
+
+@interface OLMKit : NSObject
+
+//! Project version string for OLMKit, the same as libolm.
++ (NSString*)versionString;
+
+@end
diff --git a/xcode/OLMKit/OLMKit.m b/xcode/OLMKit/OLMKit.m
index e7bfd25..c383650 100644
--- a/xcode/OLMKit/OLMKit.m
+++ b/xcode/OLMKit/OLMKit.m
@@ -19,11 +19,15 @@
 
 #include "olm/olm.h"
 
-NSString *OLMKitVersionString()
+@implementation OLMKit
+
++ (NSString*)versionString
 {
     uint8_t major, minor, patch;
-
+    
     olm_get_library_version(&major, &minor, &patch);
-
+    
     return [NSString stringWithFormat:@"%tu.%tu.%tu", major, minor, patch];
 }
+
+@end