Improving code style.  Adding docs