Updated docs and made  some style adjustments