Add register read/write support