Update get_state function