Added text classification and additional argument to the argument
parser.