didnt test but looks fine

formatter/formatter.go

@@ -1,20 +1,31 @@
 package formatter
 
-import "fmt"
+import (
+	"fmt"
+	"strings"
+)
+
+func Cube(n int) int {
+	return n*n
+}
 
 const (
     Reset   = "\033[0m"
-    Red     = "\033[31m"
-    Green   = "\033[32m"
-    Yellow  = "\033[33m"
-    Blue    = "\033[34m"
-    Magenta = "\033[35m"
-    Cyan    = "\033[36m"
+    C    = "\033[36m"
+    M = "\033[35m"
+    Y  = "\033[33m"
+    R     = "\033[31m"
+    G   = "\033[32m"
+    B    = "\033[34m"
 )
 
 func Format(op string, in, out int) string {
     return fmt.Sprintf("%s%s%s of %s%d%s = %s%d%s",
-        Blue, op, Reset,
-        Yellow, in, Reset,
-        Green, out, Reset)
+        B, op, Reset,
+        Y, in, Reset,
+        G, out, Reset)
+}
+
+func Trim(s string) string {
+  return strings.TrimSpace(s)
 }

main.go

@@ -4,11 +4,15 @@ import (
 	"fmt"
 	"os"
 
+	"flag"
 	"tech-interview/calculator"
 	"tech-interview/formatter"
 )
 
 func main() {
+	cube := flag.Bool("cube", false, "print cube")
+	flag.Parse()
+
     if len(os.Args) < 2 {
         fmt.Println("Usage: orders <number>")
         os.Exit(1)
@@ -19,8 +23,14 @@ func main() {
         fmt.Println("Invalid number:", os.Args[1])
         os.Exit(1)
     }
-
+  fmt.Println(*cube) // TODO: remove before committing!
+    if *cube {
+    cu := formatter.Cube(n)
+    out := formatter.Format("cube", n, cu)
+    fmt.Println(out)
+    } else {
     sq := calculator.Square(n)
     out := formatter.Format("square", n, sq)
     fmt.Println(out)
+    }
 }