added __future__ import command to estimate_probs function