added initial probabilities and Theta class