diff --git a/src/main.rs b/src/main.rs
index 72b8d07..a7de592 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,11 +1,14 @@
 extern crate baichat_rs;
 
-use baichat_rs::ask;
+use baichat_rs::ThebAI;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
-    let mut response = ask("Hello, world!").await?;
+    let mut ai = ThebAI::new(None);
+    
+
+    let mut response = ai.ask("Hello, world!").await?;
 
 
     println!("{:?}", response);