diff --git a/src/main/java/net/onrc/onos/util/GraphDBOperation.java b/src/main/java/net/onrc/onos/util/GraphDBOperation.java
index 1d5ab20..e097bf9 100644
--- a/src/main/java/net/onrc/onos/util/GraphDBOperation.java
+++ b/src/main/java/net/onrc/onos/util/GraphDBOperation.java
@@ -271,4 +271,8 @@
 	public void rollback() {
 		conn.endTx(Transaction.ROLLBACK);
 	}
+
+	public void close() {
+		conn.close();
+	}
 }
diff --git a/src/main/java/net/onrc/onos/util/IDBOperation.java b/src/main/java/net/onrc/onos/util/IDBOperation.java
index 2eeb527..1a8571d 100644
--- a/src/main/java/net/onrc/onos/util/IDBOperation.java
+++ b/src/main/java/net/onrc/onos/util/IDBOperation.java
@@ -38,5 +38,8 @@
 	public Iterable<IFlowEntry> getAllFlowEntries();
 	public void removeFlowEntry(IFlowEntry flowEntry);
 	
-	public IDBConnection getDBConnection();
+	public IDBConnection getDBConnection();	
+	public void commit();
+	public void rollback();
+	public void close();
 }