diff --git a/EmptyDataSet-Swift.podspec b/EmptyDataSet-Swift.podspec
index dc2020d..b5eb852 100644
--- a/EmptyDataSet-Swift.podspec
+++ b/EmptyDataSet-Swift.podspec
@@ -16,7 +16,7 @@ Pod::Spec.new do |s|
   #
 
   s.name         = "EmptyDataSet-Swift"
-  s.version      = "1.0.0"
+  s.version      = "4.0.0"
   s.summary      = "DZNEmptyDataSet with Swift."
 
   # This description is used to generate tags and improve search results.
@@ -91,7 +91,7 @@ Pod::Spec.new do |s|
   #  Not including the public_header_files will make all headers public.
   #
 
-  s.source_files  = "EmptyDataSet/*"
+  s.source_files  = "EmptyDataSet-Swift/EmptyDataSet/*"
 
   # s.public_header_files = "Classes/**/*.h"
 
diff --git a/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetDelegate.swift b/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetDelegate.swift
index a52e0e5..6e36c50 100644
--- a/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetDelegate.swift
+++ b/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetDelegate.swift
@@ -90,7 +90,7 @@ public protocol EmptyDataSetDelegate: class {
 
 }
 
-extension EmptyDataSetDelegate {
+public extension EmptyDataSetDelegate {
     
     func emptyDataSetShouldFadeIn(_ scrollView: UIScrollView) -> Bool {
         return true
diff --git a/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetSource.swift b/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetSource.swift
index e9c54fd..42a6b72 100644
--- a/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetSource.swift
+++ b/EmptyDataSet-Swift/EmptyDataSet/EmptyDataSetSource.swift
@@ -101,7 +101,7 @@ public protocol EmptyDataSetSource: class {
 
 }
 
-extension EmptyDataSetSource {
+public extension EmptyDataSetSource {
     
     func title(forEmptyDataSet scrollView: UIScrollView) -> NSAttributedString? {
         return nil
diff --git a/EmptyDataSet-Swift/OriginalUsageViewController.swift b/EmptyDataSet-Swift/OriginalUsageViewController.swift
index 50a25e6..bf34d03 100644
--- a/EmptyDataSet-Swift/OriginalUsageViewController.swift
+++ b/EmptyDataSet-Swift/OriginalUsageViewController.swift
@@ -334,7 +334,6 @@ class OriginalUsageViewController: UITableViewController, EmptyDataSetSource, Em
         return animation;
     }
     
-    
     func buttonTitle(forEmptyDataSet scrollView: UIScrollView, for state: UIControlState) -> NSAttributedString? {
         var text: String?
         var font: UIFont?
@@ -391,7 +390,7 @@ class OriginalUsageViewController: UITableViewController, EmptyDataSetSource, Em
         }
         return NSAttributedString.init(string: text!, attributes: attributes)
     }
-    
+ 
     func buttonBackgroundImage(forEmptyDataSet scrollView: UIScrollView, for state: UIControlState) -> UIImage? {
         var imageName = "button_background_\(application["display_name"]!)".lowercased()