Adding e2e tests, not currently working