add tests
integration tests