Returned comment, fixed other comments, moved method emit_probs