added testing and training visualization