add e2e test
add more tests