Tweaks for text sanitization and better modelling