Add e2e tests