Cleanup train and changed dataset