diff --git a/src/lib/FilmPredictor.mjs b/src/lib/FilmPredictor.mjs
index 4a653fb..7e134e0 100644
--- a/src/lib/FilmPredictor.mjs
+++ b/src/lib/FilmPredictor.mjs
@@ -4,6 +4,8 @@ import path from 'path';
 import fs from 'fs';
 
 import tf from '@tensorflow/tfjs-node';
+// import tf from '@tensorflow/tfjs-node-gpu';
+// import tf from '@tensorflow/tfjs';
 
 import genres from './Genres.mjs';
 
diff --git a/src/lib/Genres.mjs b/src/lib/Genres.mjs
index 7910608..32f5ca2 100644
--- a/src/lib/Genres.mjs
+++ b/src/lib/Genres.mjs
@@ -1,9 +1,9 @@
 "use strict";
 
 export default [
-	"Action",
-	"Adventure",
-	"Animation",
+	"Action",		// 0
+	"Adventure",	// 1
+	"Animation",	// 2
 	"Biography",
 	"Comedy",
 	"Crime",
@@ -23,5 +23,5 @@ export default [
 	"Sport",
 	"Thriller",
 	"War",
-	"Western"
+	"Western"		// 22
 ];