removed probability features because they were hurting performance