Minor adjustments.
Split out fit and transform in order to reuse the code when running the model for new emails.