Adding in integration tests