Update dataset creation to use CLI