Added a flag to input training file.