Add remove argument