Improving the binary classification model