Added working step method for latent state in transmission model