added remove() method and relevant tests