use all models for plotting, no set_trace anymore