Need multi column filter data from st.aggrid

dear boss I have observed that my code executes correctly; however, it only returns a single column value. I require two columns: “name” and “fee.” I attempted to modify the code to include multiple column names, but encountered an error. Please review my code and provide the correct version to retrieve data from two columns in st.aggrid.

this is my code
import streamlit as st
import pandas as pd
from st_aggrid import AgGrid
df = pd.DataFrame({‘Name’: [‘A’, ‘B’, ‘C’], ‘Age’: [25, 24, 1], ‘Fee’: [225, 224, 221]})
if “names_active” not in st.session_state:
st.session_state[“names_active”] = list(df[‘Name’])

return_df = AgGrid(data = df, data_return_mode = ‘FILTERED’)
st.session_state[“names_active”] = list(return_df[‘data’][‘Name’, ‘fee’]) # this give error
#st.session_state[“names_active”] = list(return_df[‘data’][‘Name’]) # this ok

st.markdown(st.session_state[“names_active”])

give only one column