Update documentation. Tweak comment and spacing