Define dataset
Prepare dataset