Restore remove behaviour like list and add remove_all