Makes mem.New return a pointer, handles page size 0 edge case

main.go

@@ -27,9 +27,20 @@ func main() {
 
 	scanner := bufio.NewScanner(file)
 
+	realMemory, err := mem.New(2048, 16)
+	if err != nil {
+		errorMessage := fmt.Sprintf("Error creating real memory: %s", err.Error())
+		panic(errorMessage)
+	}
+
+	swapMemory, err := mem.New(2048, 16)
+	if err != nil {
+		errorMessage := fmt.Sprintf("Error creating swap memory: %s", err.Error())
+		panic(errorMessage)
+	}
 	comp = cpu.New(
-		mem.New(2048, 16),
-		mem.New(1024, 16),
+		*realMemory,
+		*swapMemory,
 	)
 
 	for scanner.Scan() {

mem/mem.go

@@ -1,5 +1,7 @@
 package mem
 
+import "errors"
+
 type page struct {
 	pid            string
 	virtualAddress int
@@ -12,12 +14,15 @@ type Memory struct {
 }
 
 // New creates a new Memory
-func New(size, pageSize int) Memory {
+func New(size, pageSize int) (*Memory, error) {
+	if pageSize == 0 {
+		return nil, errors.New("PageSize should not be zero")
+	}
 	numOfPages := size / pageSize
-	return Memory{
+	return &Memory{
 		pages:    make([]page, numOfPages),
 		PageSize: pageSize,
-	}
+	}, nil
 }
 
 // AllocatePage ...