diff --git a/logic/nodes.go b/logic/nodes.go
index 1e2727aa..29caa405 100644
--- a/logic/nodes.go
+++ b/logic/nodes.go
@@ -19,7 +19,8 @@ import (
 	"golang.org/x/crypto/bcrypt"
 )
 
-const RELAY_NODE_ERR = "could not find relay for node "
+// RELAY_NODE_ERR - error to return if relay node is unfound
+const RELAY_NODE_ERR = "could not find relay for node"
 
 // GetNetworkNodes - gets the nodes of a network
 func GetNetworkNodes(network string) ([]models.Node, error) {
@@ -508,7 +509,7 @@ func GetNodeRelay(network string, relayedNodeAddr string) (models.Node, error) {
 			}
 		}
 	}
-	return relay, errors.New(RELAY_NODE_ERR + relayedNodeAddr)
+	return relay, errors.New(RELAY_NODE_ERR + " " + relayedNodeAddr)
 }
 
 // GetNodeByIDorMacAddress - gets the node, if a mac address exists, but not id, then it should delete it and recreate in DB with new ID
diff --git a/logic/relay.go b/logic/relay.go
index 8b6e2b97..3af4e008 100644
--- a/logic/relay.go
+++ b/logic/relay.go
@@ -84,6 +84,7 @@ func SetRelayedNodes(yesOrno string, networkName string, addrs []string) error {
 	return nil
 }
 
+// SetNodeIsRelayed - Sets IsRelayed to on or off for relay
 func SetNodeIsRelayed(yesOrno string, id string) error {
 	node, err := GetNodeByID(id)
 	if err != nil {
@@ -106,6 +107,7 @@ func SetNodeIsRelayed(yesOrno string, id string) error {
 	return database.Insert(node.ID, string(data), database.NODES_TABLE_NAME)
 }
 
+// PeerListUnRelay - call this function if a relayed node fails to get its relay: unrelays node and gets new peer list
 func PeerListUnRelay(id string, network string) ([]models.Node, error) {
 	err := SetNodeIsRelayed("no", id)
 	if err != nil {