code_generator: Add get_registers