diff --git a/hw2/app.py b/hw2/app.py
index fc986c5..ed8139c 100644
--- a/hw2/app.py
+++ b/hw2/app.py
@@ -2,6 +2,8 @@ from langchain import hub
 from langchain.agents import AgentExecutor, create_react_agent, load_tools
 from langchain.tools import tool
 from langchain_openai import ChatOpenAI
+from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
+
 from dotenv import load_dotenv
 from tools import lookup_ip, lookup_name
 from langsmith import Client
@@ -20,7 +22,7 @@ load_dotenv()
 #client = Client()
 llm = ChatOpenAI(model_name="gpt-4-turbo", temperature=0)
 
-tools = load_tools(["serpapi", "terminal"], allow_dangerous_tools=True)
+tools = load_tools(["serpapi", "terminal", "dalle-image-generator"], allow_dangerous_tools=True)
 tools.extend([lookup_name, lookup_ip])
 
 base_prompt = hub.pull("langchain-ai/react-agent-template")