Refactor adding args to CustomYamlAssistant class to a separate method