moved visualizing the model later