Added integration tests
Setup integration tests