Changed updateDisease function, still doesn't work