Fixed off by 1 error in setting class weights.