squash code fix
squash docs/tests