Fixed add/update function args