Extend script to yeild dataset,dbs,prediction