Added model object for parameter and updated the flow