Changing Disease Model a bit to get the correct disease Name