add tests, fix bug
erase and tests added