Add function to save markov chains (temporary)