Figured out how to add arg correctly