adding integration tests + assertions