Updating Distribution chapter figure code, round 2