Added word identity feature, made the training function to print model information.