Improve process flow to get initial model