add possibility of using multiple initial states