Clean up display restoration for editReg
Add edit user registers functions.