correction

changed remove  method;
added some tests