Added visualization tool. Error recorded is now MSE in training.