Added prepare dataset