--- a/heuristics/historyHeuristics.php
+++ b/heuristics/historyHeuristics.php
@@ -20,11 +20,12 @@
 $agencyTransactions = Array();
 function getAgencyTransactions($agencyName)
 {
-	global $agencyTransactions;
+	global $agencyTransactions,$conn;
 	if (!$agencyTransactions[$agencyName]) {
-		$query = 'select count(*) from contractnotice where agencyName = "' . $agencyName . '"';
-		$result = mysql_query($query);
-		$r = mysql_fetch_array($result, MYSQL_BOTH);
+		$query = 'select count(*) from contractnotice where "agencyName" = \'' . pg_escape_string ($agencyName) . '"\'';
+    $result = $conn->query($query);
+    //echo $query;
+    $r = $result->fetch(PDO::FETCH_BOTH);
 		$agencyTransactions[$agencyName] = $r[0];
 	}
 	return $agencyTransactions[$agencyName];
@@ -49,11 +50,11 @@
 }
 function getStatsAgencyTransactions()
 {
-	global $averageAgencyTransactions, $stddevAgencyTransactions;
-	$query = "select avg(count), STDDEV(count) from (select count(*) as count
-                from contractnotice group by agencyName) as a;";
-	$result = mysql_query($query);
-	$r = mysql_fetch_array($result, MYSQL_BOTH);
+	global $averageAgencyTransactions, $stddevAgencyTransactions,$conn;
+	$query = 'select avg(count), STDDEV(count) from (select count(*) as count
+                from contractnotice group by "agencyName") as a;';
+    $result = $conn->query($query);
+    $r = $result->fetch(PDO::FETCH_BOTH);
 	$averageAgencyTransactions = $r[0];
 	$stddevAgencyTransactions = $r[1];
 }
@@ -78,20 +79,21 @@
 $supplierTransactions = Array();
 function getSupplierTransactions($supplierName, $supplierABN)
 {
-	global $supplierTransactions;
+	global $supplierTransactions,$conn;
 	if ($supplierABN != 0 && $supplierABN != "") {
 		if (!$supplierTransactions[$supplierABN]) {
-			$query = 'select count(*) from contractnotice where supplierABN = "' . $supplierABN . '"';
-			$result = mysql_query($query);
-			$r = mysql_fetch_array($result, MYSQL_BOTH);
+			$query = "select count(*) from contractnotice where \"supplierABN\" = '" . $supplierABN . "'";
+			    $result = $conn->query($query);
+    $r = $result->fetch(PDO::FETCH_BOTH);
 			$supplierTransactions[$supplierABN] = $r[0];
 		}
 		return $supplierTransactions[$supplierABN];
 	}
 	if (!$supplierTransactions[$supplierName]) {
-		$query = 'select count(*) from contractnotice where supplierName = "' . $supplierName . '"';
-		$result = mysql_query($query);
-		$r = mysql_fetch_array($result, MYSQL_BOTH);
+		$query = "select count(*) from contractnotice where \"supplierName\" = '" . $supplierName . "'";
+                
+    $result = $conn->query($query);
+    $r = $result->fetch(PDO::FETCH_BOTH);
 		$supplierTransactions[$supplierName] = $r[0];
 	}
 	return $supplierTransactions[$supplierName];
@@ -116,10 +118,12 @@
 }
 function getStatsSupplierTransactions()
 {
-	global $averageSupplierTransactions, $stddevSupplierTransactions;
-	$query = 'select avg(count), stddev(count) from (select IF(supplierABN != "",supplierABN,supplierName) as supplierID, count(*) as count from contractnotice group by supplierID) as a;';
-	$result = mysql_query($query);
-	$r = mysql_fetch_array($result, MYSQL_BOTH);
+	global $averageSupplierTransactions, $stddevSupplierTransactions,$conn;
+	$query = 'select avg(count), stddev(count) from (
+            select (case when "supplierABN" != 0 THEN "supplierABN"::text ELSE "supplierName" END) as "supplierID",
+            count(*) as count from contractnotice group by "supplierID") as a;';
+    $result = $conn->query($query);
+    $r = $result->fetch(PDO::FETCH_BOTH);
 	$averageSupplierTransactions = $r[0];
 	$stddevSupplierTransactions = $r[1];
 }