diff --git a/tetris-ai-robertko.js b/tetris-ai-robertko.js
index c23a827..5ac877d 100644
--- a/tetris-ai-robertko.js
+++ b/tetris-ai-robertko.js
@@ -203,13 +203,7 @@ function rankShafts(brickId, dropLocation, tempBoard) {
 	return rank;
 }
 
-function chooseDropLocation(brickId) {
-	var dropLocations = getAllPossibleDropLocations(brickId);
-	
-	if (dropLocations.length == 0) {
-		return null;
-	}
-
+function rank(brickId, dropLocations) {
 	var bestRank = 0;
 	var bestId = -1;
 
@@ -238,6 +232,20 @@ function chooseDropLocation(brickId) {
 		}
 	}
 
+	return [ bestId, bestRank ];
+}
+
+function chooseDropLocation(brickId) {
+	var dropLocations = getAllPossibleDropLocations(brickId);
+	
+	if (dropLocations.length == 0) {
+		return null;
+	}
+
+	var best = rank(brickId, dropLocations);
+	var bestId = best[0];
+	var bestRank = best[1];
+
 	return dropLocations[bestId];
 }