ENH: update tests + example