removed the 0 prob threshold... this was what was making it crash?