added disease function