refactor properties again and visualizeLearner ---> plotLearnerPrediction