Updating Distribution chapter figure code, round 1