Added constant correction. Generalized code for data, number of fates and sites.