Cleanup model, now specify the data number