Removed unnecessary code for ModelRefine