Add more tests
Release v0.4.0