Added updateDisease function, doesn't work