Added basic integration test