From 5e19f376fb28b7cbd9429c5cf1582630a2643ee7 Mon Sep 17 00:00:00 2001
From: Pedro Albanese <68971450+pedroalbanese@users.noreply.github.com>
Date: Fri, 30 Apr 2021 18:35:37 -0300
Subject: [PATCH] Update main.go

---
 main.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 617e93a..7e7b830 100644
--- a/main.go
+++ b/main.go
@@ -22,15 +22,15 @@ import (
 func main() {
     flag.Parse()
 
-        if (len(os.Args) < 2) {
-	fmt.Println("SHA3 Keccak Hashsum Tool - ALBANESE Lab (c) 2020-2021\n")
+
+        if (len(os.Args) < 2) || (*bits != 224 && *bits != 256 && *bits != 384 && *bits != 512) {
+	fmt.Println("SHA3 Hashsum Tool - ALBANESE Lab (c) 2020-2021\n")
 	fmt.Println("Usage of",os.Args[0]+":")
         fmt.Printf("%s [-v] [-b N] [-c <hash.ext>] [-r] -t <file.ext>\n\n", os.Args[0])
         flag.PrintDefaults()
         os.Exit(1)
         }
 
-
         if *target != "" && *recursive == false {
 	files, err := filepath.Glob(*target)
 	if err != nil {