Adding Integration test