Fix some warnings

Update probability updating