dont forget the tokenizer