added more code to plug in the model

still doesn't work!