Change code to fit the Mega