Small prob adjustments
Fixes for NEAT