Update registration functionality