first version that reproduce binary classification demo