code for genetic diversity figure