Some updates to user register feature