Improve the docs for store_initial_state

@sarina Say yes please!