Incorporating habitat distribution in wildlife disease models