Fixed similar nil bug as last commit
Updated walks for new probability