Added simple integration test