Get e2e tests working (#6)