For a 2 state model and a 3 state model, add an example where we compute and plot the partial heat capacities, population fractions, and state transition terms.