Added a bit more to text clasification stuff