diff --git a/streamlit_app.py b/streamlit_app.py
index 5512ca7..2da3f2a 100644
--- a/streamlit_app.py
+++ b/streamlit_app.py
@@ -285,8 +285,8 @@ def play_old_messages(container):
     #     disabled=model in st.session_state['rqa'] and st.session_state['rqa'][model].memory is None)
 
 left_column, right_column = st.columns([1, 1])
-right_column = right_column.container(height=600, border=False)
-left_column = left_column.container(height=600, border=False)
+right_column = right_column.container(border=True)
+left_column = left_column.container(border=True)
 
 with right_column:
     uploaded_file = st.file_uploader(
@@ -483,5 +483,6 @@ def generate_color_gradient(num_elements):
             input=st.session_state['binary'],
             annotation_outline_size=2,
             annotations=st.session_state['annotations'],
-            render_text=True
+            render_text=True,
+            height=600
         )