Added integration tests
Adding integration tests