diff --git a/Docs/NAT-Traveral.md b/Docs/NAT-Traveral.md
index 4508a46..f9a69a3 100644
--- a/Docs/NAT-Traveral.md
+++ b/Docs/NAT-Traveral.md
@@ -35,6 +35,3 @@ func main() {
 }
 ```
 
-// todo diagram representation of the interaction with 
-nodes spawned in the p2p network.
-
diff --git a/abstractions/init.go b/abstractions/init.go
index 4fdae6f..06a4d2c 100644
--- a/abstractions/init.go
+++ b/abstractions/init.go
@@ -1,27 +1,27 @@
 package abstractions
 
 import (
-    "github.com/Akilan1999/p2p-rendering-computation/config"
-    "github.com/Akilan1999/p2p-rendering-computation/config/generate"
-    "github.com/Akilan1999/p2p-rendering-computation/server"
-    "github.com/gin-gonic/gin"
+	"github.com/Akilan1999/p2p-rendering-computation/config"
+	"github.com/Akilan1999/p2p-rendering-computation/config/generate"
+	"github.com/Akilan1999/p2p-rendering-computation/server"
+	"github.com/gin-gonic/gin"
 )
 
 // Init Initialises p2prc
-func Init(name string) (config *config.Config, err error) {
-    // set the config file with default paths
-    config, err = generate.SetDefaults(name, false)
-    if err != nil {
-        return
-    }
-    return
+func Init(name string, customConfig interface{}) (config *config.Config, err error) {
+	// set the config file with default paths
+	config, err = generate.SetDefaults(name, false, customConfig, false)
+	if err != nil {
+		return
+	}
+	return
 }
 
 // Start p2prc in a server mode
 func Start() (*gin.Engine, error) {
-    engine, err := server.Server()
-    if err != nil {
-        return nil, err
-    }
-    return engine, nil
+	engine, err := server.Server()
+	if err != nil {
+		return nil, err
+	}
+	return engine, nil
 }