Added integration tests
Added integration tests