Code shuffle, need to get updateDisease working