A digital correlator upgrade for the Arcminute MicroKelvin Imager