Adding Integration Tests