Add e2e_test.go