Adding the sbox to the python model.
(1) Fixed constant names. (2) Added variable allocation for round keys.