Changed the training inputs to X and y instead of samples and labels to be generic.