Refactoring "twoClass" to "classification"