Almost ready. Still need work, use mltk's tokenizer and maybe change a little bit the structure