ArrayList types changed to Disease