Add handling for script arguments.