In the future, more novels should be run in this model to keep updating the word probability. In most of the country, rating system of novels has at least three classes: G, PG-13 and A. However, the model right now just includes two rating level. Thus, the model also needs an update to distinguish more rating levels.