Adding sample registers for data in and number of rounds since we need them stable during the whole block.