Update

app.py

@@ -21,8 +21,10 @@ with st.spinner("Loading data...", show_time=True):
 
 if "problem" in st.query_params:
     problem_id = int(st.query_params["problem"])
-    st.write(f"Problem [{problem_id}]")
     problem_instance = problem_dict[problem_id]
+
+    with st.container():
+        st.write(f"Problem [{problem_id}]")
     
     with st.expander("Problem Description"):
         st.markdown(problem_instance["question_content"])