Clean up dataset, and train new model
Clean up dataset, and train new model