Add AA.remove() return behaviour