Added method to store the Thetas after calculation