improve bench script