ENH: more tests + streamline example