Adding more state register update control. Adding round counter logic.