Add AMI registration code.