updated with new initial states