finished adding e2e tests, modified remove method