Remove references to unnecessary avgProb