tiny change added: 2-category test dataset