diff --git a/mach_override_example.xcworkspace/contents.xcworkspacedata b/mach_override_example.xcworkspace/contents.xcworkspacedata
index b0ef05d..c79c36c 100644
--- a/mach_override_example.xcworkspace/contents.xcworkspacedata
+++ b/mach_override_example.xcworkspace/contents.xcworkspacedata
@@ -5,7 +5,7 @@
       location = "group:objc_runtime_hooker/objc_runtime_hooker.xcodeproj">
    </FileRef>
    <FileRef
-      location = "absolute:/Users/richardross/Documents/projects/mach_override_example/mach_override/mach_override.xcodeproj">
+      location = "group:mach_override/mach_override.xcodeproj">
    </FileRef>
    <FileRef
       location = "group:mach_override/../mach_override_test_app/mach_override_test_app.xcodeproj">
diff --git a/objc_runtime_hooker/objc_runtime_hooker/objc_runtime_hooker.cpp b/objc_runtime_hooker/objc_runtime_hooker/objc_runtime_hooker.cpp
index 2a13ff6..1277549 100644
--- a/objc_runtime_hooker/objc_runtime_hooker/objc_runtime_hooker.cpp
+++ b/objc_runtime_hooker/objc_runtime_hooker/objc_runtime_hooker.cpp
@@ -20,18 +20,34 @@
 // looking at the order of the 'link binary with libraries' phase.
 __attribute__((constructor))
 static void _hook_objc_runtime() {
-  kern_return_t err;
-  MACH_OVERRIDE(void, method_exchangeImplementations, (Method m1, Method m2), &err) {
-    printf("Exchanging implementations for method %s and %s.\n", sel_getName(method_getName(m1)), sel_getName(method_getName(m2)));
+    kern_return_t err;
+    MACH_OVERRIDE(void, method_exchangeImplementations, (Method m1, Method m2), &err) {
+        printf("Exchanging implementations for method %s and %s.\n", sel_getName(method_getName(m1)), sel_getName(method_getName(m2)));
+        
+        method_exchangeImplementations_reenter(m1, m2);
+    }
+    END_MACH_OVERRIDE(method_exchangeImplementations);
+    
+    MACH_OVERRIDE(void, method_setImplementation, (Method method, IMP imp), &err) {
+        printf("Setting new implementation for method %s.\n", sel_getName(method_getName(method)));
+        
+        method_setImplementation_reenter(method, imp);
+    }
+    END_MACH_OVERRIDE(method_setImplementation);
+    
+    MACH_OVERRIDE(void, class_replaceMethod, (Class cls, SEL name, IMP imp, const char *types), &err) {
+        printf("Setting new implementation for method %s of class%s.\n", sel_getName(name), class_getName(cls));
+        
+        class_replaceMethod_reenter(cls, name, imp, types);
+    }
+    END_MACH_OVERRIDE(class_replaceMethod);
+    
+    MACH_OVERRIDE(void, class_addMethod, (Class cls, SEL name, IMP imp, const char *types), &err) {
+        printf("Adding new for method %s of class%s.\n", sel_getName(name), class_getName(cls));
+        
+        class_addMethod_reenter(cls, name, imp, types);
+    }
+    END_MACH_OVERRIDE(class_addMethod);
+    
 
-    method_exchangeImplementations_reenter(m1, m2);
-  }
-  END_MACH_OVERRIDE(method_exchangeImplementations);
-
-  MACH_OVERRIDE(void, method_setImplementation, (Method method, IMP imp), &err) {
-    printf("Setting new implementation for method %s.\n", sel_getName(method_getName(method)));
-
-    method_setImplementation_reenter(method, imp);
-  }
-  END_MACH_OVERRIDE(method_setImplementation);
-}
\ No newline at end of file
+}