added plot in training part; added red indicator for wrong testing score