Add e2e coverage