add add_states tests