Rework the code to use set_register() with the string name of the desired register to set. Add a get_register() method. Signed-off-by: Kevin O'Connor <kevin@koconnor.net>