Added function to generate a better initial model state