Start adding integration tests