Add tests
2.0.1