fix the feature to set the image directory in argument
mention the feature to set the image directory in first argument