Add ruby code to get args