Added training on data sampled from Jan's model