Adding round counter logic.
Adding logic to select the number of rounds to generate keys for.