Change logic to improve handling of rows updated

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/notebooks/Controller.ts

@@ -87,8 +87,8 @@ export class IBMiController {
               const results = await query.run();
 
               const table = results.data;
-              if (table === undefined && results.success && !results.has_results && results.update_count > 0) {
-                items.push(vscode.NotebookCellOutputItem.text(`Statement executed successfully. ${results.update_count} rows affected.`, `text/markdown`));
+              if (table === undefined && results.success && !results.has_results) {
+                items.push(vscode.NotebookCellOutputItem.text(`Statement executed successfully. ${results.update_count ? `${results.update_count} rows affected.` : ``}`, `text/markdown`));
                 break;
               }