change the code to run school dataset