Updating Distribution chapter figure code, round 3