Mihir | Adding integration tests, and tear down method