Added creating confusion matrix to main and adjusted how numbers to train on are passed in to creating sample and validating.