revise disc_model, edit main.cc