more tests and better example docstring yay.