refactored data_prob.py. It's pretty now.