finalized script for school dataset