Some style fixes + documentation