Added a function log_probability to data_prob.py