Add more integration tests

Closes #6.