diff --git a/lib/features/accounts/screens/account_statement_screen.dart b/lib/features/accounts/screens/account_statement_screen.dart
index 3e2c16e..83b5119 100644
--- a/lib/features/accounts/screens/account_statement_screen.dart
+++ b/lib/features/accounts/screens/account_statement_screen.dart
@@ -293,10 +293,20 @@ class _AccountStatementScreen extends State<AccountStatementScreen> {
                                     : '',
                                 style: const TextStyle(fontSize: 12),
                               ),
-                              trailing: Text(
-                                "₹${tx.amount}",
-                                style: const TextStyle(fontSize: 17),
-                              ),
+                             trailing: Column(
+  crossAxisAlignment: CrossAxisAlignment.end,
+  mainAxisAlignment: MainAxisAlignment.center,
+  children: [
+    Text(
+      "₹${tx.amount}",
+      style: const TextStyle(fontSize: 17),
+    ),
+    Text(
+      "Bal: ₹${tx.balance}",
+      style: const TextStyle(fontSize: 12), // Style matches tx.name
+    ),
+  ],
+),                                                                   
                               onTap: () {
                                 Navigator.push(
                                   context,
diff --git a/lib/main.dart b/lib/main.dart
index a62f353..ad4c61c 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -17,13 +17,13 @@ void main() async {
   ]);
 
   // Check for device compromise
-  // final compromisedMessage = await SecurityService.deviceCompromisedMessage;
-  // if (compromisedMessage != null) {
-  //   runApp(MaterialApp(
-  //     home: SecurityErrorScreen(message: compromisedMessage),
-  //   ));
-  //   return;
-  // }
+  final compromisedMessage = await SecurityService.deviceCompromisedMessage;
+  if (compromisedMessage != null) {
+    runApp(MaterialApp(
+      home: SecurityErrorScreen(message: compromisedMessage),
+    ));
+    return;
+  }
 
   // Initialize dependencies
   await setupDependencies();