Better model (needs training), precision and recall, correct train and test set