Add option to merge env states instead of update
Document env.merge_states
Set current env.state