Perfect.

Could you add some unit tests for your pull request ?