diff --git a/08/cnn/cnn.js b/08/cnn/cnn.js index 493102f..3ac438b 100644 --- a/08/cnn/cnn.js +++ b/08/cnn/cnn.js @@ -28,6 +28,12 @@ // Find categories that this page is part of var categories = findCategories(data); + if ( categories.length > 0 && categories.length < 2){ + //if we have only one category we add another one at random + var keys = Object.keys(data) + keys.splice(categories[1], 1) + categories.push(keys[Math.floor(Math.random() * keys.length)]) + } // Select a pseudo-randomised category for buttonA var buttonA = categories[Math.floor(Math.random() * categories.length)];