diff --git a/docs/_build/objects.inv b/docs/_build/objects.inv
index f0411be..29f98ec 100644
Binary files a/docs/_build/objects.inv and b/docs/_build/objects.inv differ
diff --git a/docs/_build/searchindex.js b/docs/_build/searchindex.js
index 7c55d8f..9fe9d4e 100644
--- a/docs/_build/searchindex.js
+++ b/docs/_build/searchindex.js
@@ -1 +1 @@
-Search.setIndex({docnames:["about","docs","generated/mlconfound","index","install","quickstart"],envversion:{"sphinx.domains.c":2,"sphinx.domains.changeset":1,"sphinx.domains.citation":1,"sphinx.domains.cpp":4,"sphinx.domains.index":1,"sphinx.domains.javascript":2,"sphinx.domains.math":2,"sphinx.domains.python":3,"sphinx.domains.rst":2,"sphinx.domains.std":2,"sphinx.ext.viewcode":1,nbsphinx:3,sphinx:56},filenames:["about.md","docs.md","generated/mlconfound.rst","index.rst","install.md","quickstart.rst"],objects:{"":{mlconfound:[2,0,0,"-"]},"mlconfound.plot":{plot_graph:[1,1,1,""],plot_null_dist:[1,1,1,""]},"mlconfound.simulate":{identity:[1,1,1,""],simulate_y_c_yhat:[1,1,1,""],sinh_arcsinh:[1,1,1,""]},"mlconfound.stats":{CptResults:[1,2,1,""],ResultsFullyConfounded:[1,2,1,""],ResultsPartiallyConfounded:[1,2,1,""],full_confound_test:[1,1,1,""],generalization_test:[1,1,1,""],partial_confound_test:[1,1,1,""]},"mlconfound.stats.CptResults":{null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_x_y:[1,3,1,""],r2_x_z:[1,3,1,""],r2_y_z:[1,3,1,""]},"mlconfound.stats.ResultsFullyConfounded":{null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_y_c:[1,3,1,""],r2_y_yhat:[1,3,1,""],r2_yhat_c:[1,3,1,""]},"mlconfound.stats.ResultsPartiallyConfounded":{null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_y_c:[1,3,1,""],r2_y_yhat:[1,3,1,""],r2_yhat_c:[1,3,1,""]},mlconfound:{plot:[1,0,0,"-"],simulate:[1,0,0,"-"],stats:[1,0,0,"-"]}},objnames:{"0":["py","module","Python module"],"1":["py","function","Python function"],"2":["py","class","Python class"],"3":["py","attribute","Python attribute"]},objtypes:{"0":"py:module","1":"py:function","2":"py:class","3":"py:attribute"},terms:{"0":[1,4,5],"00":5,"00814":[0,3,5],"01":5,"017867":5,"027":5,"03":5,"031732":5,"03758519":1,"039042":5,"03955979":1,"03998411":1,"067903":5,"09613765":1,"1":[0,1,4,5],"100":[1,5],"1000":[1,5],"11":4,"17":4,"175":[0,1],"187028":5,"197":[0,1],"2":[0,1,4,5],"2009":1,"2020":[0,1],"2021":[0,3,5],"210914":5,"2111":[0,3,5],"22579272":1,"237854":5,"3":1,"30471708":1,"31057339":1,"32193037":1,"323":5,"35013318":1,"4":[1,4],"42":[1,5],"48801839":1,"5":[1,5],"50":[1,5],"50886059":1,"58it":5,"595":5,"6":1,"61":4,"64it":5,"7":1,"732287":5,"7504512":1,"76":5,"761":1,"770":1,"780":1,"786173":5,"82":[0,1],"83888754":1,"88854985":1,"8900947":1,"9":1,"95":1,"96":1,"case":5,"class":1,"default":[1,5],"export":0,"float":1,"function":[1,5],"import":3,"int":1,"null":[0,1,5],"return":[1,5],"true":[1,5],"while":1,A:[1,5],As:0,For:5,In:5,No:5,The:[0,1,5],To:1,_:[0,1],ab:0,about:[1,3,4],accord:5,accur:5,across:5,addit:1,adjust:5,affect:1,ag:5,against:5,al:1,alia:1,all:[1,5],allow:1,also:5,altern:5,an:[1,5],analyz:3,appli:5,applic:1,approach:1,ar:[0,1,5],arcsinh:1,argumnet:1,array_lik:1,arxiv:[0,3,5],associ:5,assumpt:0,attribut:1,avail:1,ax:1,b:[0,1],back:[0,1,4],bar:1,barber:0,base:[1,5],batch:1,being:5,berrett:[0,1],between:[1,5],bia:[0,1,3,5],bias:[0,5],bin_c:1,bin_i:1,bin_yhat:1,binomi:[1,5],biometrika:1,bool:1,build:0,built:5,c:[1,3,5],c_i:1,c_name:[1,5],callabl:1,can:[1,5],carlo:[1,5],cat_c:1,cat_i:1,cat_yhat:1,categor:1,center:1,chain:[1,5],chang:1,character:5,ci:5,classif:1,coeffici:1,com:[3,4],comparison:5,cond_dist_method:1,condit:[0,1,5],condition:5,confid:[1,5],confoudn:[0,1],confound:[0,3],confound_test_result:1,contain:1,contribut:[1,5],control:[0,1],core:4,cornel:0,correct:5,correl:1,could:5,counfound:3,cptresult:1,creat:5,custom:5,data:[1,5],datafram:5,dataset:5,de:0,decent:5,delta:1,depend:[1,5],depict:1,describ:1,detail:[1,3],determin:1,develop:0,dict:1,direct:5,directli:[1,5],discoveri:5,displai:0,distribut:[0,1,5],document:[0,4],doe:5,dot2tex:4,dot:1,driven:[1,5],e:[1,5],edu:0,effect:[0,1,3],entir:1,epsilon:1,especi:5,essen:0,estim:[1,5],et:1,evid:5,exampl:[1,5],expect:1,explain:[1,5],explicit:5,extens:1,extrem:5,f:[0,1],fals:[1,5],fdr:5,feedback:[0,1,4],field:[0,1],figur:1,file:1,flag:[1,5],follow:5,form:5,framework:0,from:[1,3,5],full:[0,5],full_confound_test:1,fulli:[0,1,5],g:[1,5],gam:1,gener:1,generaliz:0,generalization_test:1,germani:0,git:4,github:[0,4],give:[0,1,4],given:[1,5],graph:[1,5],graphic:5,graphviz:[1,4],h0:1,h0_c:5,h0_y:5,h0_yhat:5,h1_c:5,h1_y:5,h1_yhat:5,hamper:0,handl:1,hat:[1,5],have:5,here:5,high:0,histogram:[1,5],histplot:1,hood:1,hospit:0,howev:5,http:[0,3,4],hypothesi:[0,1,5],i:[1,5],ident:1,implement:0,impli:5,increas:5,independ:[0,1,5],independencewhil:0,index:1,indic:5,input:1,instal:0,interv:1,introduc:1,investig:5,involv:1,ipynb:1,iq:5,issu:[0,4],its:1,j:0,job:1,joblib:4,jone:1,journal:[0,1],kurtosi:1,kwarg:1,lab:[0,3,4],label:5,lack:0,lb:0,learn:[1,3,5],let:5,librari:0,limit:5,line:5,linear:1,logit:1,low:[1,5],lower:5,m:1,machin:[1,3,5],mailto:0,main:[0,1,4],mani:[0,5],markov:[1,5],mathcal:1,matplotlib:1,mcmc_step:1,mean:1,method:1,methodolog:[0,1],might:5,mlconfound:[4,5],model:[0,1,3,5],modul:3,mont:[1,5],more:[1,3,5],multinomi:1,multipl:5,must:[1,5],n:[1,5],n_job:[1,5],name:[1,5],namedtupl:1,ndarrai:1,necessarili:5,next:5,non:[0,1],none:1,nonlin_trf_fun:1,normal:[0,1],note:[1,5],notebook:1,now:5,null_distribut:1,num_perm:[1,5],number:[1,5],numer:1,numpi:1,object:1,observ:[1,5],often:[0,5],onli:1,option:[4,5],out:1,outfile_bas:[1,5],output:[0,1],p:[1,5],p_ci:[1,5],packag:0,page:[0,1,3,4],panda:5,parallel:1,param:1,paramet:1,parametr:[0,5],partial:0,partial_confound_test:[1,3,5],pass:1,path:1,pd:5,perform:[1,5],permut:[0,1,5],pewsei:1,pip:[3,4],plot:[4,5],plot_graph:[1,5],plot_null_dist:[1,5],pni:[0,3,4],posit:1,power:0,precis:1,predict:[0,1,3,5],prep:1,preprint:[0,3,5],pretti:5,print:[1,5],probabl:1,process:5,processor:1,progress:1,propos:[0,5],provid:5,purpos:[1,5],put:5,py:1,pypi:3,quantif:[0,3,5],quantifi:3,quickstart:[0,1,3,4],r2:5,r2_x_y:1,r2_x_z:1,r2_y_c:1,r2_y_yhat:1,r2_y_z:1,r2_yhat_c:1,r:[0,1,5],random:[1,5],random_st:[1,5],rate:5,recent:0,recommend:1,red:5,refer:1,relat:1,represent:5,reproduc:5,research:[0,5],respect:[0,5],result:[1,5],resultsfullyconfound:1,resultspartiallyconfound:1,ret:5,return_null_dist:[1,5],rigor:0,robust:0,royal:[0,1],s:[1,5],sampl:1,samworth:0,save:1,scale:1,seaborn:[1,4,5],search:[1,3],see:1,seed:5,seri:[0,1],set:[0,1,5],sex:1,should:1,show:5,shown:0,signal:1,signific:[1,5],significantli:0,sim:1,simpli:[1,5],simul:0,simulate_y_c_yhat:[1,5],sinh:1,sinh_archsin:1,sinh_arcsinh:1,skew:1,sn:5,so:5,societi:[0,1],sole:5,some:5,sourc:1,spisak:[0,1,3,5],squar:1,stat:[3,5],state:1,statist:[0,1,3,5],statsmodel:4,step:[1,5],still:5,str:1,stronger:5,t:0,tama:[0,1],target:[1,5],techniqu:5,teh:1,test:0,test_fully_confound:1,test_partially_confound:1,than:[1,5],thei:5,theoret:0,theori:0,therefor:1,thi:[1,5],third:1,thoma:1,tool:3,tqdm:4,transform:1,transfrom:1,tupl:1,two:5,uk:0,uncondit:5,undepend:1,under:1,unit:1,univers:0,unpermut:5,upper:5,us:[1,5],valid:[0,1],valu:[1,5],variabl:[1,3,5],varianc:1,w_:1,w_cyhat:[1,5],w_yc:[1,5],w_yyhat:[1,5],wang:0,we:5,weight:1,what:5,whether:[1,5],without:1,x:1,y:[0,1,3,5],y_i:1,y_name:[1,5],yhat:[1,3],yhat_nam:[1,5],zero:[1,5]},titles:["About mlconfound","Documentation ","mlconfound","Welcome to the docs of \u2018mlconfound\u2019! ","Install","Quickstart"],titleterms:{"import":5,about:0,also:[0,4],author:0,bleed:4,bug:[0,4],citat:3,confound:[1,5],contact:[0,4],content:1,doc:3,document:[1,3],edg:4,exampl:3,from:4,full:1,github:3,h0:5,h1:5,index:3,indic:1,instal:[3,4],mlconfound:[0,1,2,3],modul:1,navig:[1,3],necessari:5,packag:5,partial:[1,5],plot:1,pypi:4,quickstart:5,recommend:4,refer:[0,5],report:[0,4],requir:4,see:[0,4],simul:[1,5],sourc:3,stabl:4,star:3,stat:1,statu:1,test:[1,5],welcom:3}})
\ No newline at end of file
+Search.setIndex({docnames:["about","docs","generated/mlconfound","index","install","quickstart"],envversion:{"sphinx.domains.c":2,"sphinx.domains.changeset":1,"sphinx.domains.citation":1,"sphinx.domains.cpp":4,"sphinx.domains.index":1,"sphinx.domains.javascript":2,"sphinx.domains.math":2,"sphinx.domains.python":3,"sphinx.domains.rst":2,"sphinx.domains.std":2,"sphinx.ext.viewcode":1,nbsphinx:3,sphinx:56},filenames:["about.md","docs.md","generated/mlconfound.rst","index.rst","install.md","quickstart.rst"],objects:{"":{mlconfound:[2,0,0,"-"]},"mlconfound.plot":{plot_graph:[1,1,1,""],plot_null_dist:[1,1,1,""]},"mlconfound.simulate":{identity:[1,1,1,""],simulate_y_c_yhat:[1,1,1,""],sinh_arcsinh:[1,1,1,""]},"mlconfound.stats":{CptResults:[1,2,1,""],ResultsFullyConfounded:[1,2,1,""],ResultsPartiallyConfounded:[1,2,1,""],full_confound_test:[1,1,1,""],generalization_test:[1,1,1,""],partial_confound_test:[1,1,1,""]},"mlconfound.stats.CptResults":{expected_r2_x_y:[1,3,1,""],null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_x_y:[1,3,1,""],r2_x_z:[1,3,1,""],r2_y_z:[1,3,1,""]},"mlconfound.stats.ResultsFullyConfounded":{expected_r2_y_yhat:[1,3,1,""],null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_y_c:[1,3,1,""],r2_y_yhat:[1,3,1,""],r2_yhat_c:[1,3,1,""]},"mlconfound.stats.ResultsPartiallyConfounded":{expected_r2_yhat_c:[1,3,1,""],null_distribution:[1,3,1,""],p:[1,3,1,""],p_ci:[1,3,1,""],r2_y_c:[1,3,1,""],r2_y_yhat:[1,3,1,""],r2_yhat_c:[1,3,1,""]},mlconfound:{plot:[1,0,0,"-"],simulate:[1,0,0,"-"],stats:[1,0,0,"-"]}},objnames:{"0":["py","module","Python module"],"1":["py","function","Python function"],"2":["py","class","Python class"],"3":["py","attribute","Python attribute"]},objtypes:{"0":"py:module","1":"py:function","2":"py:class","3":"py:attribute"},terms:{"0":[1,4,5],"00":5,"00814":[0,3,5],"01":5,"017867":5,"024":5,"027":5,"03":5,"031732":5,"03758519":1,"039042":5,"03955979":1,"03998411":1,"04":5,"044":5,"05":1,"059":5,"065":5,"067903":5,"09613765":1,"1":[0,1,4,5],"100":[1,5],"1000":[1,5],"11":4,"17":4,"175":[0,1],"187028":5,"197":[0,1],"2":[0,1,4,5],"2009":1,"2020":[0,1],"2021":[0,3,5],"210914":5,"2111":[0,3,5],"22579272":1,"237854":5,"3":1,"30471708":1,"31057339":1,"32193037":1,"323":5,"35013318":1,"4":[1,4],"42":[1,5],"48801839":1,"5":[1,5],"50":[1,5],"50886059":1,"58it":5,"595":5,"6":1,"61":4,"64it":5,"7":1,"732287":5,"7504512":1,"76":5,"761":1,"770":1,"780":1,"786173":5,"82":[0,1],"83888754":1,"88854985":1,"8900947":1,"9":1,"95":[1,5],"96":1,"case":5,"class":1,"default":[1,5],"export":0,"float":1,"function":[1,5],"import":3,"int":1,"null":[0,1,5],"return":[1,5],"true":[1,5],"while":1,A:[1,5],As:0,For:5,In:5,No:5,The:[0,1,5],To:1,_:[0,1],ab:0,about:[1,3,4],accord:5,accur:5,across:5,addit:1,adjust:5,affect:1,ag:5,against:5,al:1,alia:1,all:[1,5],allow:1,also:5,altern:5,an:[1,5],analyz:3,appli:5,applic:1,approach:1,ar:[0,1,5],arcsinh:1,argumnet:1,array_lik:1,arxiv:[0,3,5],associ:5,assumpt:0,attribut:1,avail:1,ax:1,b:[0,1],back:[0,1,4],bar:1,barber:0,base:[1,5],batch:1,being:5,berrett:[0,1],between:[1,5],bia:[0,1,3,5],bias:[0,5],bin_c:1,bin_i:1,bin_yhat:1,binomi:[1,5],biometrika:1,bool:1,build:0,built:5,c:[1,3,5],c_i:1,c_name:[1,5],callabl:1,can:[1,5],carlo:[1,5],cat_c:1,cat_i:1,cat_yhat:1,categor:1,center:1,chain:[1,5],chang:1,character:5,ci:5,classif:1,coeffici:[1,5],com:[3,4],comparison:5,cond_dist_method:1,condit:[0,1,5],condition:5,confid:[1,5],confoudn:[0,1],confound:[0,3],confound_test_result:1,contain:1,contribut:[1,5],control:[0,1],core:4,cornel:0,correct:5,correl:1,could:5,counfound:3,cptresult:1,creat:[],custom:5,data:[1,5],datafram:[],dataset:5,de:0,decent:5,delta:1,depend:[1,5],depict:1,describ:1,detail:[1,3],determin:[1,5],develop:0,dict:1,direct:5,directli:[1,5],discoveri:5,displai:0,distribut:[0,1,5],document:[0,4],doe:5,dot2tex:4,dot:1,driven:[1,5],e:[1,5],edu:0,effect:[0,1,3],entir:1,epsilon:1,especi:5,essen:0,estim:[1,5],et:1,evid:5,exampl:[1,5],expect:[1,5],expected_r2_x_i:1,expected_r2_y_yhat:1,expected_r2_yhat_c:1,explain:[1,5],explicit:5,extens:1,extrem:5,f:[0,1],fall:5,fals:[1,5],fdr:5,feedback:[0,1,4],field:[0,1],figur:1,file:1,flag:[1,5],follow:5,form:5,framework:0,from:[1,3,5],full:[0,5],full_confound_test:1,fulli:[0,1,5],g:[1,5],gam:1,gener:1,generaliz:0,generalization_test:1,germani:0,git:4,github:[0,4],give:[0,1,4],given:[1,5],graph:[1,5],graphic:5,graphviz:[1,4],h0:1,h0_c:5,h0_y:5,h0_yhat:5,h1_c:5,h1_y:5,h1_yhat:5,hamper:0,handl:1,hat:[1,5],have:5,here:5,high:0,histogram:[1,5],histplot:1,hood:1,hospit:0,howev:5,http:[0,3,4],hypothesi:[0,1,5],i:[1,5],ident:1,implement:0,impli:5,increas:5,independ:[0,1,5],independencewhil:0,index:1,indic:5,input:1,instal:0,interv:1,introduc:1,investig:5,involv:1,ipynb:1,iq:5,issu:[0,4],its:1,j:0,job:1,joblib:4,jone:1,journal:[0,1],kurtosi:1,kwarg:1,lab:[0,3,4],label:5,lack:0,lb:0,learn:[1,3,5],let:5,librari:0,limit:5,line:5,linear:1,logit:1,low:[1,5],lower:5,m:1,machin:[1,3,5],mailto:0,main:[0,1,4],mani:[0,5],markov:[1,5],mathcal:1,matplotlib:1,mcmc_step:1,mean:1,method:1,methodolog:[0,1],might:5,mlconfound:[4,5],model:[0,1,3,5],modul:3,mont:[1,5],more:[1,3,5],multinomi:1,multipl:5,must:[1,5],n:[1,5],n_job:[1,5],name:[1,5],namedtupl:1,ndarrai:1,necessarili:5,next:5,non:[0,1],none:1,nonlin_trf_fun:1,normal:[0,1],note:[1,5],notebook:1,now:5,null_distribut:1,num_perm:[1,5],number:[1,5],numer:1,numpi:1,object:1,observ:[1,5],often:[0,5],onli:1,option:[4,5],our:5,out:1,outfile_bas:[1,5],output:[0,1],outsid:5,p:[1,5],p_ci:[1,5],packag:0,page:[0,1,3,4],panda:5,parallel:1,param:1,paramet:1,parametr:[0,5],partial:0,partial_confound_test:[1,3,5],pass:1,path:1,pd:5,percentil:5,perform:[1,5],permut:[0,1,5],pewsei:1,pip:[3,4],plot:[4,5],plot_graph:[1,5],plot_null_dist:[1,5],pni:[0,3,4],posit:1,power:0,precis:1,predict:[0,1,3,5],prep:1,preprint:[0,3,5],pretti:[],print:1,probabl:1,process:5,processor:1,progress:1,propos:[0,5],provid:5,purpos:[1,5],put:5,py:1,pypi:3,quantif:[0,3,5],quantifi:3,quantil:1,quickstart:[0,1,3,4],r2:5,r2_x_y:1,r2_x_z:1,r2_y_c:1,r2_y_yhat:1,r2_y_z:1,r2_yhat_c:1,r:[0,1,5],random:[1,5],random_st:[1,5],rang:5,rate:5,recent:0,recommend:1,red:5,refer:1,reject:5,relat:1,represent:5,reproduc:5,research:[0,5],respect:[0,5],result:[1,5],resultsfullyconfound:1,resultspartiallyconfound:1,ret:5,return_null_dist:[1,5],rigor:0,robust:0,royal:[0,1],s:[1,5],sampl:1,samworth:0,save:1,scale:1,seaborn:[1,4,5],search:[1,3],see:1,seed:5,seri:[0,1],set:[0,1,5],sex:1,should:1,show:5,shown:0,signal:1,signific:[1,5],significantli:0,sim:1,simpli:[1,5],simul:0,simulate_y_c_yhat:[1,5],sinh:1,sinh_archsin:1,sinh_arcsinh:1,skew:1,sn:5,so:5,societi:[0,1],sole:5,some:5,sourc:1,spisak:[0,1,3,5],squar:1,stat:[3,5],state:1,statist:[0,1,3,5],statsmodel:4,step:[1,5],still:5,str:1,stronger:5,t:0,tama:[0,1],target:[1,5],techniqu:5,teh:1,test:0,test_fully_confound:1,test_partially_confound:1,than:[1,5],thei:5,theoret:0,theori:0,therefor:1,thi:[1,5],third:1,thoma:1,tool:3,tqdm:4,transform:1,transfrom:1,tupl:1,two:5,uk:0,uncondit:5,uncounfoudn:5,undepend:1,under:1,unit:1,univers:0,unpermut:5,upper:5,us:[1,5],valid:[0,1],valu:[1,5],variabl:[1,3,5],varianc:1,w_:1,w_cyhat:[1,5],w_yc:[1,5],w_yyhat:[1,5],wang:0,we:5,weight:1,well:5,what:5,whether:[1,5],within:5,without:1,x:1,y:[0,1,3,5],y_i:1,y_name:[1,5],yhat:[1,3],yhat_nam:[1,5],zero:[1,5]},titles:["About mlconfound","Documentation ","mlconfound","Welcome to the docs of \u2018mlconfound\u2019! ","Install","Quickstart"],titleterms:{"import":5,about:0,also:[0,4],author:0,bleed:4,bug:[0,4],citat:3,confound:[1,5],contact:[0,4],content:1,doc:3,document:[1,3],edg:4,exampl:3,from:4,full:1,github:3,h0:5,h1:5,index:3,indic:1,instal:[3,4],mlconfound:[0,1,2,3],modul:1,navig:[1,3],necessari:5,packag:5,partial:[1,5],plot:1,pypi:4,quickstart:5,recommend:4,refer:[0,5],report:[0,4],requir:4,see:[0,4],simul:[1,5],sourc:3,stabl:4,star:3,stat:1,statu:1,test:[1,5],welcom:3}})
\ No newline at end of file
diff --git a/docs/quickstart.rst b/docs/quickstart.rst
index 20e111a..1e1a754 100644
--- a/docs/quickstart.rst
+++ b/docs/quickstart.rst
@@ -60,8 +60,7 @@ The random seed is set for reproducible results. The flag
``return_null_dist`` is set so that the full permutation-based null
distribution is returned, e.g. for plotting purposes.
-The pandas dataframe is solely created for “pretty-printing” the
-results.
+
.. code:: ipython3
@@ -101,8 +100,10 @@ results.
ci lower |
ci upper |
R2(y,c) |
- R2(y^,c) |
R2(y,y^) |
+ Expected R2(y^,c) |
+ Observed R2(y^,c) |
+
|