improved visualizeLearner, not really done