Register Wrapper improvements

added a wrapper for registers