Remove cleaning of the training set, breaks all the fucking time