Code now handles disease names.