Code for SAB dataset