Improve code a bit

socket.go

@@ -2,19 +2,18 @@ package main
 
 import (
 	"bufio"
+	"fmt"
 	"net"
-	"os"
 )
 
-func runServer() {
+func runServer() error {
 	serverSocket, err := net.Listen("tcp", "127.0.0.1:7777")
 	if err != nil {
-		logger.Errorf("Could not create listening socket: %s\n", err.Error())
+		myError := fmt.Errorf("could not create listening socket: %s", err.Error())
-		os.Exit(1)
+		return myError
-	} else {
+	}
 	logger.Infof("Now listening for incoming connections.\n")
 	defer serverSocket.Close()
-	}
 	for {
 		clientConnection, err := serverSocket.Accept()
 		if err != nil {