Small changes to scripts
Changed special case code to use task set size