Made the updateDisease function really messy, kind of works but not really