Update CategoriseAlgorithm.java