added a remove method + comments