updates dataset creation code