Update registration function