Prepare dataset
Train model