diff --git a/WebServer/Components/Pages/Home.razor b/WebServer/Components/Pages/Home.razor
index 4c54c54f..d12dc1d1 100644
--- a/WebServer/Components/Pages/Home.razor
+++ b/WebServer/Components/Pages/Home.razor
@@ -19,19 +19,28 @@
 </div>
 
 <div class="main-area">
-    <button @onclick="pullData">@buttonText</button>
+    <button @onclick="pullandtrain">@button1Text</button>
+    <button @onclick="predict">@button2Text</button>
 </div>
 
 
 @code {
 
-    string buttonText = "Pull finance data";
+    string button1Text = "Train AI";
+    string button2Text = "Predict AI";
 
-    async Task pullData(){
-        buttonText = "Do not refresh the page. The data is refreshing.";
+    async Task pullandtrain(){
+        button1Text = "Do not refresh the page. The data is refreshing.";
         await Task.Delay(1);
-        aiModule.PullData();
-        buttonText = "Refresh Completed";
+        aiModule.TrainAI();
+        button1Text = "Refresh Completed";
+    }
+
+    async Task predict(){
+        button1Text = "Do not refresh the page. The data is refreshing.";
+        await Task.Delay(1);
+        aiModule.TrainAI();
+        button1Text = "Refresh Completed";
     }
 
 }
\ No newline at end of file
diff --git a/WebServer/Program.cs b/WebServer/Program.cs
index 867a7961..da72e72f 100644
--- a/WebServer/Program.cs
+++ b/WebServer/Program.cs
@@ -10,7 +10,7 @@ using DataBase;
 #endif
 
 if (args.Contains("Pull-Stock-Data")) {
-    interopModule.PullData();
+    interopModule.TrainAI();
 } else {
     // Create the webapp
     var builder = WebApplication.CreateBuilder(args);