diff --git a/code_blocks/tools/paywalls_3_new.swift b/code_blocks/tools/paywalls_3_new.swift
index 4210ad2dc..9f0ea0ad0 100644
--- a/code_blocks/tools/paywalls_3_new.swift
+++ b/code_blocks/tools/paywalls_3_new.swift
@@ -12,6 +12,18 @@ struct App: View {
             .sheet(isPresented: self.$displayPaywall) {
                 // We handle scroll views for you, no need to wrap this in a ScrollView
                 PaywallView() 
+                    .onPurchaseCompleted { customerInfo in
+                        print("Purchase completed: \(customerInfo.entitlements)")
+                    }
+                     .onPurchaseFailure { error in
+                        print("Purchase error: \(error.localizedDescription)")
+                    }
+                    .onRestoreCompleted { customerInfo in
+                        print("Restore completed: \(customerInfo.entitlements)")
+                    }
+                    .onRestoreFailure { error in
+                        print("Purchase error: \(error.localizedDescription)")
+                    }
             }
     }
-} 
\ No newline at end of file
+}