Add text pre-processing + improve text classification