Update app.py

app.py

@@ -590,13 +590,18 @@ elif st.session_state.current_index == len(st.session_state.data):
 else:
     st.write(f"Thank you for taking part in this study!")
 
-# Navigation buttons
-if st.session_state.current_index > 0:
-    if st.button("Previous"):
-        navigate(-1)
+col1, col2, col3 = st.columns([2, 2, 1.5])
 
+# Navigation buttons
+with col1:
+    if st.session_state.current_index > 0:
+        if st.button("Previous"):
+            navigate(-1)
+with col3:
+    if st.button("Last edited instance"):
+        go_to_index = get_start_index()
+        navigate_to_row(go_to_index)
 
-    
 if 0 <= st.session_state.current_index < len(st.session_state.data):
     st.write(f"Page {st.session_state.current_index + 1} out of {len(st.session_state.data)}")
 
@@ -605,20 +610,18 @@ def navigate_to_index():
         go_to_index = int(st.session_state.go_to)
         # Check if the index is within the valid range
         if 0 <= go_to_index < len(st.session_state.data):
+            st.session_state.go_to = 0
             navigate_to_row(go_to_index)
         else:
             st.info(f"{go_to_index} is not in the range of data.")
     except ValueError:
         st.error("The page number must be an integer.")
 
-buff, col1, col2 = st.columns([2, 2, 1.5])
-with col2:
+with col3:
     with st.form(key="nav_form"):
         st.number_input("Go to Index: ", key="go_to", format="%.0f")
         st.form_submit_button("Go", on_click=navigate_to_index)
-    if st.button("Last edited instance"):
-        go_to_index = get_start_index()
-        navigate_to_row(go_to_index)
+