Updated unit tests
Updated e2e tests