Start refactoring register handling to be a bit simpler
Add support for setting registers in the new and cool way