Add tests

Create unit and integration tests.