Made changes to Integration Test Class