Add function to count arguments