Add new e2e tests