Better docstring
Adding a basic test. No warranty it will pass