X-Git-Url: https://git.r.bdr.sh/rbdr/lyricli/blobdiff_plain/a968bff7e21c9372deca5fce5921999b45d5d348..169297c8e8464ca0cf600e9e426f11d170ee169d:/Sources/main.swift

diff --git a/Sources/main.swift b/Sources/main.swift
index e4b2760..42d4855 100644
--- a/Sources/main.swift
+++ b/Sources/main.swift
@@ -21,6 +21,21 @@ func createParser() -> ([String:Option], CommandLineKit) {
 
     parser.addOptions(Array(flags.values))
 
+    parser.formatOutput = {parseString, type in
+
+        var formattedString: String
+
+        switch(type) {
+        case .About:
+            formattedString = "\(parseString) [<artist_name> <song_name>]"
+            break
+        default:
+            formattedString = parseString
+        }
+
+        return parser.defaultFormat(formattedString, type: type)
+    }
+
     return (flags, parser)
 }