Add E2E test