print usage when a symbol is not given

1 files changed, 5 insertions, 8 deletions

diff --git a/main.go b/main.go
index e2102e5..5c03cf2 100644
--- a/main.go
+++ b/main.go
@@ -31,14 +31,11 @@ func Visit(n *html.Node) {
 }
 
 func main() {
-	response, err1 := http.Get("https://finance.yahoo.com/quote/MSFT/")
-	if err1 != nil {
-		os.Exit(1)
-	}
-
-	doc, err := html.Parse(response.Body)
-	if err != nil {
-		os.Exit(2)
+	if len(os.Args) != 2 {
+		fmt.Printf("Usage:\n\t%s <symbol>\n", os.Args[0])
+		return
 	}
+	response, _ := http.Get(fmt.Sprintf("https://finance.yahoo.com/quote/%s/", os.Args[1]))
+	doc, _ := html.Parse(response.Body)
 	Visit(doc)
 }