A minor refactor to improve efficiency once I add a new classification algorithm.