bug fixes + documentation