Added update_states test