added e2e tests, improve ui