Add get_state