Finished writing the training portion of the code.