diff --git a/01_basic_response.py b/01_basic_response.py
index 665b397..68e4577 100644
--- a/01_basic_response.py
+++ b/01_basic_response.py
@@ -1,5 +1,18 @@
+import os
+
+import google.generativeai as genai
+
+
 def main():
-    print("Hello from mini-ema!")
+    api_key = os.environ.get("GEMINI_API_KEY")
+    if not api_key:
+        raise EnvironmentError("Set the GEMINI_API_KEY environment variable to call Gemini API.")
+    genai.configure(api_key=api_key)
+
+    model = genai.GenerativeModel("gemini-1.5-flash")
+    prompt = "Write a story about a magic backpack."
+    response = model.generate_content(prompt)
+    print(response.text)
 
 
 if __name__ == "__main__":
diff --git a/pyproject.toml b/pyproject.toml
index 2ddb7e8..7a7491a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,4 +4,6 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = []
+dependencies = [
+    "google-generativeai==0.8.3",
+]