Added functions for plotting covariate balance. Made returns explicit. Cleaned up documenation