add the model visualization