Added test set (hold out data); added a few data visualizations.