Add E2E tests