Add tests
Release 0.4.0