I like the idea of specific-purpose toy models. How did you tune the code and what dataset you used?