Updated to use register.