Added vapnik penalty, and fixed a bug in the plot routine.