diff --git a/src/codeas/ui/components/deployment_ui.py b/src/codeas/ui/components/deployment_ui.py
index e07699c..e356da9 100644
--- a/src/codeas/ui/components/deployment_ui.py
+++ b/src/codeas/ui/components/deployment_ui.py
@@ -1,8 +1,11 @@
 import streamlit as st
 
-from codeas.core.state import state  # Add this import
+from codeas.core.state import state
 from codeas.use_cases.deployment import define_deployment, generate_deployment
 
+DEPLOYMENT_STRATEGY_FILE = "deployment_strategy.json"
+TERRAFORM_CODE_FILE = "terraform_code.json"
+
 
 def display():
     use_previous_outputs_strategy = st.toggle(
@@ -17,7 +20,7 @@ def display():
         with st.spinner("Defining deployment requirements..."):
             if use_previous_outputs_strategy:
                 try:
-                    previous_output = state.read_output("deployment_strategy.json")
+                    previous_output = state.read_output(DEPLOYMENT_STRATEGY_FILE)
                     st.session_state.outputs["deployment_strategy"] = type(
                         "Output",
                         (),
@@ -25,7 +28,7 @@ def display():
                             "response": {"content": previous_output["content"]},
                             "cost": previous_output["cost"],
                             "tokens": previous_output["tokens"],
-                            "messages": previous_output["messages"],  # Add this line
+                            "messages": previous_output["messages"],
                         },
                     )
                 except FileNotFoundError:
@@ -49,9 +52,9 @@ def display():
                             ].tokens,
                             "messages": st.session_state.outputs[
                                 "deployment_strategy"
-                            ].messages,  # Add this line
+                            ].messages,
                         },
-                        "deployment_strategy.json",
+                        DEPLOYMENT_STRATEGY_FILE,
                     )
             else:
                 st.session_state.outputs["deployment_strategy"] = define_deployment()
@@ -67,9 +70,9 @@ def display():
                         ].tokens,
                         "messages": st.session_state.outputs[
                             "deployment_strategy"
-                        ].messages,  # Add this line
+                        ].messages,
                     },
-                    "deployment_strategy.json",
+                    DEPLOYMENT_STRATEGY_FILE,
                 )
 
     if st.button("Preview", key="preview_deployment_strategy"):
@@ -110,7 +113,7 @@ def display_generate_deployment():
             ].response["content"]
             if use_previous_outputs_deployment:
                 try:
-                    previous_output = state.read_output("terraform_code.json")
+                    previous_output = state.read_output(TERRAFORM_CODE_FILE)
                     st.session_state.outputs["terraform_code"] = type(
                         "Output",
                         (),
@@ -118,7 +121,7 @@ def display_generate_deployment():
                             "response": {"content": previous_output["content"]},
                             "cost": previous_output["cost"],
                             "tokens": previous_output["tokens"],
-                            "messages": previous_output["messages"],  # Add this line
+                            "messages": previous_output["messages"],
                         },
                     )
                 except FileNotFoundError:
@@ -138,9 +141,9 @@ def display_generate_deployment():
                             "tokens": st.session_state.outputs["terraform_code"].tokens,
                             "messages": st.session_state.outputs[
                                 "terraform_code"
-                            ].messages,  # Add this line
+                            ].messages,
                         },
-                        "terraform_code.json",
+                        TERRAFORM_CODE_FILE,
                     )
             else:
                 st.session_state.outputs["terraform_code"] = generate_deployment(
@@ -156,9 +159,9 @@ def display_generate_deployment():
                         "tokens": st.session_state.outputs["terraform_code"].tokens,
                         "messages": st.session_state.outputs[
                             "terraform_code"
-                        ].messages,  # Add this line
+                        ].messages,
                     },
-                    "terraform_code.json",
+                    TERRAFORM_CODE_FILE,
                 )
 
     if st.button("Preview", key="preview_terraform_code"):
@@ -183,4 +186,4 @@ def display_generate_deployment():
             )
             with st.expander("Messages", expanded=False):
                 st.json(output.messages)
-            st.markdown(output.response["content"])
+            st.markdown(output.response["content"])
\ No newline at end of file