Added the test set.
Fixed graphing code.
Added mse to the test set calc.